vision0.121

app.py

@@ -5,7 +5,7 @@ from PIL import Image
 import requests
 
 # Load the pre-trained ResNet model
-model = models.mobilenet_v2(pretrained=True)
+model = models.resnet50(pretrained=True)
 model.eval()
 
 # Define the transformation for input images
@@ -40,7 +40,7 @@ def classify_image(input_image):
     return predicted_label
 
 # Gradio UI components
-image_input = gr.Image(preprocessing_fn=lambda img: Image.open(img.name))
+image_input = gr.Image()
 output_label = gr.Textbox()
 
 # Gradio interface

requirements.txt

@@ -1,5 +1,5 @@
 gradio
-pillow
+Pillow
 torch
 requests
 torchvision==0.12.0