Update app.py

app.py

@@ -1,7 +1,9 @@
 import streamlit as st
 from huggingface_hub import HfApi
 import os
+import urllib.request
 import subprocess
+import tarfile
 
 HF_TOKEN = st.secrets.get("HF_TOKEN") or os.environ.get("HF_TOKEN")
 HF_USERNAME = (
@@ -9,18 +11,32 @@ HF_USERNAME = (
     or os.environ.get("HF_USERNAME")
     or os.environ.get("SPACE_AUTHOR_NAME")
 )
-TRANSFORMERS_REPOSITORY_URL = "https://github.com/xenova/transformers.js"
-TRANSFORMERS_REPOSITORY_REVISION = "2.16.0"
+
+TRANSFORMERS_REPOSITORY_REVISION = "v3"
+TRANSFORMERS_REPOSITORY_URL = f"https://github.com/xenova/transformers.js/archive/refs/heads/{TRANSFORMERS_REPOSITORY_REVISION}.tar.gz"
 TRANSFORMERS_REPOSITORY_PATH = "./transformers.js"
+ARCHIVE_PATH = f"./transformers_{TRANSFORMERS_REPOSITORY_REVISION}.tar.gz"
 HF_BASE_URL = "https://huggingface.co"
 
 if not os.path.exists(TRANSFORMERS_REPOSITORY_PATH):
-    os.system(f"git clone {TRANSFORMERS_REPOSITORY_URL} {TRANSFORMERS_REPOSITORY_PATH}")
-
-os.system(
-    f"cd {TRANSFORMERS_REPOSITORY_PATH} && git checkout {TRANSFORMERS_REPOSITORY_REVISION}"
-)
-
+    # Download the .tar.gz file
+    print(f"Downloading the repository from {TRANSFORMERS_REPOSITORY_URL}...")
+    urllib.request.urlretrieve(TRANSFORMERS_REPOSITORY_URL, ARCHIVE_PATH)
+    
+    # Extract the .tar.gz file
+    print(f"Extracting the archive {ARCHIVE_PATH}...")
+    with tarfile.open(ARCHIVE_PATH, "r:gz") as tar:
+        tar.extractall()
+    
+    # Rename the extracted folder to match the expected path
+    # GitHub strips the leading 'v', so we handle that here
+    extracted_folder = f"./transformers.js-{TRANSFORMERS_REPOSITORY_REVISION.lstrip('v')}"
+    os.rename(extracted_folder, TRANSFORMERS_REPOSITORY_PATH)
+    
+    # Remove the downloaded .tar.gz archive
+    os.remove(ARCHIVE_PATH)
+    print("Repository downloaded and extracted successfully.")
+    
 st.write("## Convert a HuggingFace model to ONNX")
 
 input_model_id = st.text_input(
@@ -62,6 +78,15 @@ if input_model_id:
                     capture_output=True,
                     text=True,
                 )
+                
+                # Log the script output
+                print("### Script Output ###")
+                print(output.stdout)
+            
+                # Log any errors
+                if output.stderr:
+                    print("### Script Errors ###")
+                    print(output.stderr)
 
             model_folder_path = (
                 f"{TRANSFORMERS_REPOSITORY_PATH}/models/{input_model_id}"
@@ -103,4 +128,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"
-                )
+                )