Update Interface

app.py

@@ -20,38 +20,35 @@ def process(img):
     img = np.array(img) / 255.0
     return np.expand_dims(img, axis=0)
 
-def predict(img):
+def predict(img, threshold):
     img = process(img)
     pred = model.predict(img)
     class_idx = np.argmax(pred)
     confidence = pred[0][class_idx]
     class_name = class_labels[class_idx]
-    if confidence < CONFIDENCE_THRESHOLD :
+
+    if confidence < threshold:
         return {
-            'label' : 'Tidak Dikenali',
-            'confidence' : float(confidence),
-            'status' : 'Unknown',
+            "label": "Tidak Dikenali",
+            "confidence": float(confidence),
+            "status": "Unknown"
         }
     else:
-        return { 
-            'label' : class_name,
-            'confidence' : float(confidence),
-            'status' : 'Ok'
+        return {
+            "label": class_name,
+            "confidence": float(confidence),
+            "status": "OK"
         }
 
 
+
 interface = gr.Interface(
     fn=predict,
     inputs=[
         gr.Image(type="pil"),
         gr.Slider(0, 1, value=0.5, label="Confidence Threshold")
     ],
-    outputs=[
-        gr.Textbox(label="Label"),
-        gr.Textbox(label="Confidence"),
-        gr.Textbox(label="Status")
-    ],
-    title="CNN Makanan Classifier"
+    outputs=gr.JSON(label="Hasil Prediksi")
 )
 
 interface.launch()