Update : Add Labels

app.py

@@ -4,6 +4,16 @@ import numpy as np
 from PIL import Image
 
 model = tf.keras.models.load_model("best_model_mobilenetv2_finetune.h5")
+class_labels = [
+    'ayam bakar','ayam goreng','bakso','bakwan',
+    'bihun','capcay','gado-gado','ikan goreng',	
+    'kerupuk',	'martabak telur','mie','nasi goreng',	
+	'nasi putih','nugget','opor ayam','pempek',
+	'rendang','roti','sate','sosis',
+	'soto','tahu','telur','tempe',
+	'tumis kangkung',
+	'udang',
+]
 
 def process(img):
     img = img.resize((224,224))
@@ -14,7 +24,9 @@ def predict(img):
     img = process(img)
     pred = model.predict(img)
     class_idx = np.argmax(pred)
-    return f"Class : {class_idx} (confidence : {pred[0][class_idx]:.2f})"
+    confidence = pred[0][class_idx]
+    class_name = class_labels[class_idx]
+    return f"{class_name} (confidence: {confidence:.2f})"
 
 interface = gr.Interface(fn=predict, inputs=gr.Image(type="pil"),outputs="text")
 interface.launch()