Update app.py

app.py

@@ -32,7 +32,8 @@ def predict(image):
     prediction = model.predict(image)  # Predict
     predicted_class = np.argmax(prediction, axis=-1)[0]  # Get the predicted class
     class_confidence = np.max(prediction, axis=-1)[0]  # Get the confidence score
-    return {"prediction": int(predicted_class), "confidence": class_confidence}
+    #return {"prediction": int(predicted_class), "confidence": class_confidence}
+    return {"prediction": int(predicted_class)}
 
 # Create a Gradio interface
 interface = gr.Interface(fn=predict, inputs="image", outputs="json")