Update app.py

app.py

@@ -44,38 +44,17 @@ os.chdir(repo_dir)
 # # Add repo directory to sys.path so Python can find modules inside it
 sys.path.append(repo_dir)
 
-# Now you can directly load and execute `app.py` from the snapshot directory
-
-# List all files in the repo_dir to check where app.py is located
-print(f"Files in downloaded repo directory: {os.listdir(repo_dir)}")
-
-# Check if the app.py file exists
-app_path = os.path.join(repo_dir, 'app.py')
-if not os.path.exists(app_path):
-    print(f"Error: {app_path} does not exist.")
-    # You can also print out subdirectories if app.py is in a subfolder
-    subdirectories = [d for d in os.listdir(repo_dir) if os.path.isdir(os.path.join(repo_dir, d))]
-    print(f"Subdirectories in repo: {subdirectories}")
-else:
-    # Load and execute `app.py`
-    spec_app = importlib.util.spec_from_file_location("*", app_path)
-    app_module = importlib.util.module_from_spec(spec_app)
-    spec_app.loader.exec_module(app_module)
-
-    # Now you can use functions from the app_module
-    result = app_module.main()
-
 
 # Download specific files (if snapshot_download wasn't used)
-# app_path = hf_hub_download(
-#     repo_id=REPO_ID,
-#     filename="app.py",
-#     repo_type=REPO_TYPE
-# )
+app_path = hf_hub_download(
+    repo_id=REPO_ID,
+    filename="app.py",
+    repo_type=REPO_TYPE
+)
 
-# # Load and execute `app.py`
-# spec_app = importlib.util.spec_from_file_location("*", app_path)
-# app_module = importlib.util.module_from_spec(spec_app)
-# spec_app.loader.exec_module(app_module)
-## Now you can use functions from utils_module
-#result = app_module.main()
+# Load and execute `app.py`
+spec_app = importlib.util.spec_from_file_location("*", app_path)
+app_module = importlib.util.module_from_spec(spec_app)
+spec_app.loader.exec_module(app_module)
+# Now you can use functions from utils_module
+result = app_module.main()