Update app.py

app.py

@@ -1,9 +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 = (
@@ -12,8 +15,10 @@ HF_USERNAME = (
     or os.environ.get("SPACE_AUTHOR_NAME")
 )
 
-TRANSFORMERS_REPOSITORY_REVISION = "v3"
-TRANSFORMERS_REPOSITORY_URL = f"https://github.com/xenova/transformers.js/archive/refs/heads/{TRANSFORMERS_REPOSITORY_REVISION}.tar.gz"
+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"
@@ -23,15 +28,18 @@ if not os.path.exists(TRANSFORMERS_REPOSITORY_PATH):
     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)
+    # Create a temporary directory for extraction
+    with tempfile.TemporaryDirectory() as tmp_dir:
+        # Extract the .tar.gz file to temp directory
+        print(f"Extracting the archive {ARCHIVE_PATH}...")
+        with tarfile.open(ARCHIVE_PATH, "r:gz") as tar:
+            tar.extractall(tmp_dir)
+        
+        # Get the extracted folder name (there should be only one)
+        extracted_folder = os.path.join(tmp_dir, os.listdir(tmp_dir)[0])
+        
+        # Move to final destination
+        os.rename(extracted_folder, TRANSFORMERS_REPOSITORY_PATH)
     
     # Remove the downloaded .tar.gz archive
     os.remove(ARCHIVE_PATH)