Update app.py

app.py

@@ -44,18 +44,33 @@ 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
+app_path = os.path.join(repo_dir, 'app.py')
 
-# 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
-)
+# Check if app.py exists in the directory
+if os.path.exists(app_path):
+    # 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)
 
-# 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()
+else:
+    print(f"Error: {app_path} does not exist.")
+
+
+# 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
+# )
+
+# # 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()