update stuff

app.py

@@ -49,6 +49,7 @@ def predict(image):
     print(out.shape)
     probabilities = F.softmax(out[0], dim=0)
     values, indices = torch.topk(probabilities, 5)
+    print(values, indices)
 
     return {LABELS[i]: v.item() for i, v in zip(indices, values)}