Remove env variables subprocess call

app.py

@@ -1,7 +1,12 @@
-import streamlit as st
-from huggingface_hub import HfApi
+import sys
 import os
+import urllib.request
 import subprocess
+import tarfile
+import tempfile
+
+import streamlit as st
+from huggingface_hub import HfApi
 
 HF_TOKEN = st.secrets.get("HF_TOKEN") or os.environ.get("HF_TOKEN")
 HF_USERNAME = (
@@ -9,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(
@@ -51,7 +67,7 @@ if input_model_id:
             with st.spinner("Converting model..."):
                 output = subprocess.run(
                     [
-                        "python",
+                        sys.executable,
                         "-m",
                         "scripts.convert",
                         "--quantize",
@@ -61,7 +77,17 @@ if input_model_id:
                     cwd=TRANSFORMERS_REPOSITORY_PATH,
                     capture_output=True,
                     text=True,
+                    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}"
@@ -103,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"
-                )
+                )