Update app.py

app.py

@@ -14,7 +14,8 @@ def predict(img):
         # Fetch the labels dynamically from the model's vocabulary
         labels = learn.dls.vocab
         # Ensure probabilities are floats
-        return {labels[i]: {float(probs[i]) for i in range(len(labels))}
+        return {labels[i]: float(probs[i]) 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}")