Update app.py

app.py

@@ -83,8 +83,8 @@ def predict_image(input):
     # Get the confidence score of the predicted class
     confidence_score = predictions[0][predicted_class_index]
     
-    # Return input image path, predicted class label, and confidence score
-    return input, {predicted_class_label: confidence_score}
+    # Return predicted class label and confidence score
+    return {predicted_class_label: confidence_score}
 
 
 image_gradio_app = gr.Interface(