Update app.py

app.py

@@ -4,7 +4,7 @@ from torchvision import transforms
 from PIL import Image
 import gradio as gr
 
-model.load_state_dict(torch.load("squeezenet.pth"))
+model = torch.load("squeezenet.pth")
 model.eval()
 
 transform = transforms.Compose([