packages

loader.py

@@ -2,7 +2,7 @@ import os
 import sys
 from huggingface_hub import hf_hub_download
 import traceback
-
+import json 
 def load_script(file_str: str):
     """
     Downloads a file from the Hugging Face Hub and ensures a symlink exists in the current directory.
@@ -102,6 +102,9 @@ def load_scripts():
     3. Read each line from the downloaded file list, where each line specifies another file to download.
     4. After downloading all files, execute the last downloaded file.
     """
+    local_config = {}
+    if os.path.exists("local.json"):
+        local_config = json.load(open("local.json"))
     file_list = os.getenv("FILE_LIST", "").strip()
     if not file_list:
         print("No 'FILE_LIST' environment variable set. Nothing to download.")
@@ -133,7 +136,7 @@ def load_scripts():
             downloaded_files.append(file_path)
 
     # Step 4: Execute the last downloaded file
-    if downloaded_files:
+    if downloaded_files and local_config.get("execute_last_script", True):
         last_file_path = downloaded_files[-1]
         print(f"Executing the last downloaded script: '{last_file_path}'")
         try:

requirements.txt

@@ -4,7 +4,6 @@ transformers
 accelerate
 safetensors
 einops
-onnxruntime-gpu
 spaces==0.19.4
 opencv-python
 pyjwt