Update app.py

app.py

@@ -69,9 +69,23 @@ def process_p_file(uploaded_file, percentage_idx, complexity_idx):
         else:
             return f"Directory {model_repo_dir} does not exist."
 
-        print("a")
+        # Debugging: Check if the directory exists and print contents
+        if os.path.exists(model_repo_dir):
+            os.chdir(model_repo_dir)
+            print(f"Changed working directory to {os.getcwd()}")
+            print(f"Directory content: {os.listdir(os.getcwd())}")  # Debugging: Check repo content
+        else:
+            print(f"Directory {model_repo_dir} does not exist.")
+            return
+
+        # Step 2: Add the cloned repo to sys.path for imports
+        if model_repo_dir not in sys.path:
+            sys.path.append(model_repo_dir)
+        
+        # Debugging: Print sys.path to ensure the cloned repo is in the path
+        print(f"sys.path: {sys.path}")
+        
         from lwm_model import LWM
-        print("b")
         device = 'cpu'
         print(f"Loading the LWM model on {device}...")
         model = LWM.from_pretrained(device=device)