Update app.py

app.py

@@ -5,8 +5,8 @@ from PIL import Image
 import gradio as gr
 
 # Load your trained ResNet-50 model
-model = models.resnet50(pretrained=False)  # Load ResNet-50 architecture
-model.load_state_dict(torch.load("model.pth"))  # Load the trained weights (.pth)
+model = models.resnet50(pretrained=False)  # Load the ResNet-50 architecture 
+model.load_state_dict(torch.load("model.pth"))  # Load the trained weights  (.pth)
 model.eval()  # Set model to evaluation mode
 
 # Define the transformation required for the input image