Update app.py

app.py

@@ -4,6 +4,7 @@ import gradio as gr
 import tensorflow as tf
 from PIL import Image
 from tensorflow.keras.models import load_model
+
 # Load the model
 model_path = 'final_teath_classifier.h5'
 model = tf.keras.models.load_model(model_path)
@@ -31,16 +32,16 @@ def predict_image(image):
         predict_label = "Carries"
     confidence = float(np.max(predictions))
     prediction_dict = {"prediction": predict_label, "confidence": confidence}
-    return json.dumps(prediction_dict, indent=2)
+    return prediction_dict
 
 # Create the interface
-input_interface = gr.Image()
-output_interface = "json"
+input_interface = gr.inputs.Image()
+output_interface = gr.outputs.JSON()
 
 iface = gr.Interface(
     fn=predict_image,
     inputs=input_interface,
-    outputs=gr.Textbox("output"))
+    outputs=output_interface)
 
 # Launch the interface
 iface.launch(share=True)