Update app.py

app.py

@@ -65,7 +65,7 @@ def load_model(checkpoint_path, d_args):
         model = Model(d_args)
         try:
             # Load checkpoint
-            checkpoint = torch.load(checkpoint_path, map_location=torch.device("cpu"))
+            checkpoint = torch.load(checkpoint_path[i], map_location=torch.device("cpu"))
             model.load_state_dict(checkpoint)
             print(f"Model_{i} loaded successfully.")
         except Exception as e: