Update app.py

app.py

@@ -5,8 +5,9 @@ import gradio as gr
 import os
 
 torch.hub.download_url_to_file("https://github.com/pytorch/hub/raw/master/images/dog.jpg", "dog.jpg")
-model = torch.load('shufflenetv2_x1-5666bf0f80.pth')
-model.eval()
+
+ model = torch.hub.load('pytorch/vision:v0.9.0', 'shufflenet_v2_x1_0', pretrained=False)
+ model.load_state_dict(torch.load('shufflenetv2_x1-5666bf0f80.pth'))
 
 os.system("wget https://raw.githubusercontent.com/pytorch/hub/master/imagenet_classes.txt")