Update app.py

app.py

@@ -54,8 +54,7 @@ def load_custom_model():
 def process_p_file(uploaded_file, percentage_idx, complexity_idx):
     capture = PrintCapture()
     sys.stdout = capture  # Redirect print statements to capture
-    print("a")
-    print("b")
+    
     try:
         model_repo_url = "https://huggingface.co/sadjadalikhani/LWM"
         model_repo_dir = "./LWM"
@@ -71,7 +70,7 @@ def process_p_file(uploaded_file, percentage_idx, complexity_idx):
             return f"Directory {model_repo_dir} does not exist."
 
         from lwm_model import LWM
-        device = 'cuda' if torch.cuda.is_available() else 'cpu'
+        device = 'cpu'
         print(f"Loading the LWM model on {device}...")
         model = LWM.from_pretrained(device=device)