Update app.py

app.py

@@ -1,11 +1,13 @@
 import torch
 
-# Check if CUDA (GPU) is available, otherwise use CPU
-device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
+# Specify the correct path to your LoRA checkpoint file
+lora_path = "./lora_model.pt"  # Replace with the correct file name
+
+# Define the device (use CPU in this case)
+device = torch.device("cpu")
 
 # Load the checkpoint
-lora_path = "./lora_checkpoint.pt"  # Path to the checkpoint file
 checkpoint = torch.load(lora_path, map_location=device)
 
-# Print the checkpoint keys to inspect the structure
+# Print the keys in the checkpoint to check the structure
 print("Checkpoint keys:", checkpoint.keys())