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()
 
@@ -15,7 +15,7 @@ with open('best_model.pth', 'wb') as f:
     f.write(model_content)
 
 # Load your custom model onto the CPU
-model = torch.load('best_model.pth', map_location=torch.device('cpu'))
+model = torch.hub.load('ultralytics/yolov5', 'custom', path='best_model.pth')
 model.eval()
 
 # Define a function that takes an image as input and uses the model for inference