Update app.py

app.py

@@ -24,7 +24,7 @@ def predict(image):
     #predictions = np.argmax(logits, axis=1)    
     #predicted_labels = [labels[prediction] for prediction in predictions]
     #print(predicted_labels[0],logits[0][predictions[0]])
-    confidences = {labels[i]: float(logits[i]) for i in range(len(labels))}
+    confidences = {labels[i]: float(logits[0][i]) for i in range(len(labels))}
     return confidences
     #return {"predictions" : predictions }