Update app.py

app.py

@@ -44,6 +44,20 @@ os.chdir(repo_dir)
 # # Add repo directory to sys.path so Python can find modules inside it
 sys.path.append(repo_dir)
 
+# Ensure the correct path to app.py
+app_path = os.path.join(repo_dir, "app.py")
+if not os.path.exists(app_path):
+    raise FileNotFoundError(f"app.py not found in {repo_dir}")
+
+# Load and execute `app.py`
+spec_app = importlib.util.spec_from_file_location("app", app_path)
+if spec_app is None or spec_app.loader is None:
+    raise ImportError(f"Could not load module from {app_path}")
+
+app_module = importlib.util.module_from_spec(spec_app)
+spec_app.loader.exec_module(app_module)
+
+
 
 # Download specific files (if snapshot_download wasn't used)
 # app_path = hf_hub_download(
@@ -52,9 +66,9 @@ sys.path.append(repo_dir)
 #     repo_type=REPO_TYPE
 # )
 
-# Load and execute `app.py`
-spec_app = importlib.util.spec_from_file_location("*", repo_dir)
-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("*", repo_dir)
+# 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()