Update app.py

app.py

@@ -32,7 +32,12 @@ def predict_image(image):
     prediction_dict = {"prediction": predict_label, "confidence": confidence}
     #return prediction_dict
     probability_good = outputs[0][0]
-    return probability_good
+    result = {
+        "prediction": "Your Teeth are Good & You Don't Need To Visit Doctor" if probability_good > 0.5 else "Your Teeth are Bad & You Need To Visit Doctor"
+    }
+
+    return result
+
 
 # Create the interface
 input_interface = gr.Image(type = "pil")