Update app.py

app.py

@@ -4,14 +4,15 @@ import gradio as gr
 from PIL import Image
 from torchvision import transforms
 
-model = torch.hub.load('pytorch/vision:v0.10.0', 'resnet18', pretrained = True).eval()
+model = torch.hub.load('pytorch/vision:v0.6.0', 'resnet18', pretrained = True).eval()
 response = requests.get("https://git.io/JJkYN")
 labels = response.text.split("\n")
 
 def get_predictions(inp):
     inp = transforms.ToTensor()(inp).unsqueeze(0)
-    predictions = torch.nn.functional.softmax(model(inp)[0], dim = 0)
-    conf = {labels[i]: predictions[i] for i in range(len(labels))}
+    with torch.no_grad():
+        predictions = torch.nn.functional.softmax(model(inp)[0], dim = 0)
+        conf = {labels[i]: float(predictions[i]) for i in range(len(labels))}
     return conf