Update app.py

app.py

@@ -8,7 +8,7 @@ def predict(inp):
     inputs = data_transforms(inp)[None]
     model.eval()
     with torch.no_grad():
-        logits = model(inputs)
+        logits = model(inputs)['logits']
         probs = torch.softmax(logits,dim=1)
         confidences = {labels[i]: probs[0][i] for i in range(num_classes)}
     return confidences