Update app.py

app.py

@@ -5,10 +5,13 @@ from torchvision.models import resnet50
 from PIL import Image
 
 # Load the model
+from huggingface_hub import hf_hub_download
+
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 model = resnet50(pretrained=False)
 model.fc = torch.nn.Linear(model.fc.in_features, 14)  # Adjust for 14 classes
-model.load_state_dict(torch.load("resnet50_model_hf.pt", map_location=device))
+model_path = hf_hub_download(repo_id="iamomtiwari/resnet50-crop-disease", filename="resnet50_model_hf.pt")
+model.load_state_dict(torch.load(model_path, map_location=device))
 model.eval()
 
 # Define image transformations