Update app.py

app.py

@@ -1,34 +1,22 @@
 import gradio as gr
 from transformers import pipeline
-from fastapi import FastAPI
 
-# Memuat model untuk klasifikasi makanan padang
+# Load the model for food classification
 food_classifier = pipeline(task="image-classification", model="mhdiqbalpradipta/minang_food_classification")
 
+def predict_food(image):
+    # Perform prediction using the model
+    result = food_classifier(images=image)[0]
 
-def prediksi_makanan(gambar):
-    # Melakukan prediksi menggunakan model 
-    hasil = food_classifier(images=gambar)[0]
+    # Save label and score
+    food_label = result['label']
+    score = result['score']
+    return f"Food: {food_label}, Score: {score:.2f}"
 
-    # Menyimpan label dan skor
-    label_makanan = hasil['label']
-    skor = hasil['score']
-
-    return f"Makanan: {label_makanan}, Skor: {skor:.2f}"
-
-# Antarmuka Gradio
+# Gradio Interface
 image_in = gr.Image(type='pil')
 label_out = "text"
-contoh_gambar = ['ayam_goreng.jpg', 'ayam_pop.jpg', 'daging_rendang.jpg', 'dendeng_batokok.jpg', 'gulai_ikan.jpg', 'gulai_tambusu.jpg', 'gulai_tunjang.jpg', 'telur_balado.jpg', 'telur_dadar.jpg']
-
-intf = gr.Interface(fn=prediksi_makanan, inputs=image_in, outputs=label_out, examples=contoh_gambar, title="Pengklasifikasi Makanan Minang", description="Unggah gambar makanan untuk mengklasifikasikannya menjadi hidangan Minang.")
-
-# if __name__ == "__main__":
-app = FastAPI()
-
-@app.get('/')
-async def root():
-    return 'Gradio app is running at /gradio', 200
+example_images = ['ayam_goreng.jpg', 'ayam_pop.jpg', 'daging_rendang.jpg', 'dendeng_batokok.jpg', 'gulai_ikan.jpg', 'gulai_tambusu.jpg', 'gulai_tunjang.jpg', 'telur_balado.jpg', 'telur_dadar.jpg']
 
-app = gr.mount_gradio_app(app, intf, path='/gradio')
-# intf.launch(share=False)
+intf = gr.Interface(fn=predict_food, inputs=image_in, outputs=label_out, examples=example_images, title="Minang Food Classifier", description="Upload an image of food to classify it into Minang dishes.")
+intf.launch(share=False);