Update app.py

app.py

@@ -44,39 +44,17 @@ os.chdir(repo_dir)
 # # Add repo directory to sys.path so Python can find modules inside it
 sys.path.append(repo_dir)
 
-# Recursively search for app.py
-app_path = None
-for root, dirs, files in os.walk(repo_dir):
-    if "app.py" in files:
-        app_path = os.path.join(root, "app.py")
-        break
 
-if app_path is None:
-    raise FileNotFoundError(f"app.py not found in {repo_dir}")
+# 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", app_path)
-if spec_app is None or spec_app.loader is None:
-    raise ImportError(f"Could not load module from {app_path}")
-
+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 app_module
+# Now you can use functions from utils_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
-# )
-
-# # 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()