Update app.py

app.py

@@ -1,9 +1,12 @@
+import sys
 import os
+import urllib.request
 import subprocess
-import sys
+import tarfile
+import tempfile
 
-from huggingface_hub import HfApi
 import streamlit as st
+from huggingface_hub import HfApi
 
 HF_TOKEN = st.secrets.get("HF_TOKEN") or os.environ.get("HF_TOKEN")
 HF_USERNAME = (
@@ -11,18 +14,29 @@ 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_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_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"
 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}"
-)
-
+    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(
@@ -63,8 +77,17 @@ if input_model_id:
                     cwd=TRANSFORMERS_REPOSITORY_PATH,
                     capture_output=True,
                     text=True,
-                    env={}
+                    env={},
                 )
+                
+                # 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}"
@@ -106,4 +129,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"
-                )
+                )