mobilenet update

app.py

@@ -2,9 +2,10 @@ import gradio as gr
 import torch
 from torchvision import models, transforms
 from PIL import Image
+import requests
 
 # Load the pre-trained ResNet model
-model = models.resnet50(pretrained=True)
+model = models.MobileNet_V3_Small_Weights(pretrained=True)
 model.eval()
 
 # Define the transformation for input images
@@ -17,7 +18,8 @@ preprocess = transforms.Compose([
 
 # Define the labels for ImageNet classes (you may need to adjust this based on your model)
 LABELS_URL = "https://raw.githubusercontent.com/anishathalye/imagenet-simple-labels/master/imagenet-simple-labels.json"
-labels = gr.utils.get_file(LABELS_URL, cache=True).read_text().splitlines()
+response = requests.get(LABELS_URL)
+labels = response.text.splitlines()
 
 # Function to perform image classification
 def classify_image(input_image):

requirements.txt

@@ -1,4 +1,5 @@
 gradio
 pillow
 torch
+requests
 torchvision