Update app.py

app.py

@@ -26,7 +26,7 @@ def suppress_stdout():
 # Load the saved model and suppress the warnings
 with suppress_stdout():
     model = ViTForImageClassification.from_pretrained('google/vit-base-patch16-224-in21k', num_labels=6)
-    model.load_state_dict(torch.load('/vit_sugarcane_disease_detection.pth'))
+    model.load_state_dict(torch.load('vit_sugarcane_disease_detection.pth'))
     model.eval()
 
 # Define the same transformation used during training
@@ -58,6 +58,6 @@ def predict_disease(image_path):
     return predicted_label
 
 # Test with a new image
-image_path = '/zoomed_Bacterial%20Blight_614.png'  # Replace with your image path
+image_path = 'zoomed_Bacterial%20Blight_614.png'  # Replace with your image path
 predicted_label = predict_disease(image_path)
 print(f'The predicted disease type is: {predicted_label}')