Update app.py

app.py

@@ -10,7 +10,7 @@ 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)
+    predictions = torch.nn.functional.softmax(model(inp)[0], dim = 0)
     conf = {labels[i]: predictions[i] for i in range(len(labels))}
     return conf