Update app.py

app.py

@@ -29,7 +29,9 @@ def predict_image(image):
         predict_label = "Clean"
     else:
         predict_label = "Carries"
-    return {"prediction": predict_label, "confidence": float(np.max(predictions))}
+    confidence = float(np.max(predictions))
+    prediction_dict = {"prediction": predict_label, "confidence": confidence}
+    return prediction_dict
 
 # Create the interface
 input_interface = gr.Image(type ='pil')