Update app.py

app.py

@@ -15,7 +15,7 @@ def predict(img):
         # Get predictions from the model
         pred, pred_idx, probs = learn.predict(img)
         # Convert probabilities to percentages and construct the output dictionary dynamically
-        return {float(labels[i]): f"{probs[i]*100:.2f}%" for i in range(len(labels))}
+        return {labels[i]: f"{float(probs[i])*100:.2f}%" for i in range(len(labels))}
     except Exception as e:
         # Log the exception and return it as an error message
         print(f"An error occurred: {e}")