Update app.py

app.py

@@ -6,7 +6,7 @@ from PIL import Image
 
 # Authenticate and download the custom model from Hugging Face Spaces
 fs = HfFileSystem()
-model_path = 'dhhd255/efficient_net_parkinsons/best_model.pth'
+model_path = 'dhhd255/main_model/best_model.pth'
 with fs.open(model_path, 'rb') as f:
     model_content = f.read()
 
@@ -14,8 +14,8 @@ with fs.open(model_path, 'rb') as f:
 with open('best_model.pth', 'wb') as f:
     f.write(model_content)
 
-# Load your custom model
-model = torch.load('best_model.pth')
+# Load your custom model onto the CPU
+model = torch.load('best_model.pth', map_location=torch.device('cpu'))
 model.eval()
 
 # Define a function that takes an image as input and uses the model for inference