Update app.py

app.py

@@ -39,12 +39,12 @@ def predict(image):
         top_probs = top_probs.detach().numpy()[0].tolist()  # Convert to list
         top_idxs = top_idxs.detach().numpy()[0].tolist()  # Convert to list
         top_classes = [model.config.id2label[idx] for idx in top_idxs]
-        result = {top_classes[i]: top_probs[i] for i in range(3)}
+        result = {top_classes[i]: float(top_probs[i]) for i in range(3)}
         logging.info("Prediction successful.")
         return result
     except Exception as e:
         logging.error("Error during prediction: %s", e)
-        return str(e)
+        return {"error": str(e)}
 
 # Create the Gradio interface
 iface = gr.Interface(