Update app.py

app.py

@@ -11,7 +11,12 @@ def predict(image):
     arr = (np.array(img, dtype=np.float32) / 127.5) - 1.0
     preds = model.predict(np.expand_dims(arr, 0))[0]
     idx = int(np.argmax(preds))
-    return {"className": CLASS_NAMES[idx], "confidence": round(float(preds[idx]) * 100, 1)}
+    return {CLASS_NAMES[i]: float(preds[i]) for i in range(len(CLASS_NAMES))}
 
-demo = gr.Interface(fn=predict, inputs=gr.Image(type="pil"), outputs="json")
-demo.launch()
+demo = gr.Interface(
+    fn=predict,
+    inputs=gr.Image(type="pil"),
+    outputs=gr.Label(num_top_classes=3),
+    title="OralScan AI"
+)
+demo.launch()