Update app.py

app.py

@@ -6,14 +6,14 @@ from PIL import Image
 import json
 
 # Load class names
-with open('deployment/class_names.json', 'r') as f:
+with open('class_names.json', 'r') as f:
     class_names = json.load(f)
 
 # Define model
 def load_model():
     model = models.resnet50(pretrained=False)
     model.fc = nn.Linear(model.fc.in_features, len(class_names))
-    checkpoint = torch.load('deployment/reptile_classifier.pth', map_location=torch.device('cpu'))
+    checkpoint = torch.load('reptile_classifier.pth', map_location=torch.device('cpu'))
     model.load_state_dict(checkpoint['model_state_dict'])
     model.eval()
     return model