Update app.py

app.py

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