Reformat file

app.py

@@ -1,9 +1,9 @@
-import sys
 import os
-import urllib.request
 import subprocess
+import sys
 import tarfile
 import tempfile
+import urllib.request
 
 import streamlit as st
 from huggingface_hub import HfApi
@@ -17,7 +17,14 @@ HF_USERNAME = (
 
 TRANSFORMERS_BASE_URL = "https://github.com/xenova/transformers.js/archive/refs"
 TRANSFORMERS_REPOSITORY_REVISION = "3.0.0"
-TRANSFORMERS_REF_TYPE = "tags" if urllib.request.urlopen(f"{TRANSFORMERS_BASE_URL}/tags/{TRANSFORMERS_REPOSITORY_REVISION}.tar.gz").getcode() == 200 else "heads"
+TRANSFORMERS_REF_TYPE = (
+    "tags"
+    if urllib.request.urlopen(
+        f"{TRANSFORMERS_BASE_URL}/tags/{TRANSFORMERS_REPOSITORY_REVISION}.tar.gz"
+    ).getcode()
+    == 200
+    else "heads"
+)
 TRANSFORMERS_REPOSITORY_URL = f"{TRANSFORMERS_BASE_URL}/{TRANSFORMERS_REF_TYPE}/{TRANSFORMERS_REPOSITORY_REVISION}.tar.gz"
 TRANSFORMERS_REPOSITORY_PATH = "./transformers.js"
 ARCHIVE_PATH = f"./transformers_{TRANSFORMERS_REPOSITORY_REVISION}.tar.gz"
@@ -25,18 +32,18 @@ HF_BASE_URL = "https://huggingface.co"
 
 if not os.path.exists(TRANSFORMERS_REPOSITORY_PATH):
     urllib.request.urlretrieve(TRANSFORMERS_REPOSITORY_URL, ARCHIVE_PATH)
-    
+
     with tempfile.TemporaryDirectory() as tmp_dir:
         with tarfile.open(ARCHIVE_PATH, "r:gz") as tar:
             tar.extractall(tmp_dir)
-        
+
         extracted_folder = os.path.join(tmp_dir, os.listdir(tmp_dir)[0])
-        
+
         os.rename(extracted_folder, TRANSFORMERS_REPOSITORY_PATH)
-    
+
     os.remove(ARCHIVE_PATH)
     print("Repository downloaded and extracted successfully.")
-    
+
 st.write("## Convert a HuggingFace model to ONNX")
 
 input_model_id = st.text_input(
@@ -79,11 +86,11 @@ if input_model_id:
                     text=True,
                     env={},
                 )
-                
+
                 # Log the script output
                 print("### Script Output ###")
                 print(output.stdout)
-            
+
                 # Log any errors
                 if output.stderr:
                     print("### Script Errors ###")
@@ -129,4 +136,4 @@ if input_model_id:
                 st.write("You can now go and view the model on HuggingFace!")
                 st.link_button(
                     f"Go to {output_model_id}", output_model_url, type="primary"
-                )
+                )