mhdiqbalpradipta commited on
Commit
0499d90
1 Parent(s): e0edd9a

create app.py

Browse files
Files changed (1) hide show
  1. app.py +27 -0
app.py ADDED
@@ -0,0 +1,27 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import gradio as gr
2
+ from transformers import pipeline
3
+ import torch
4
+
5
+ # Memuat model untuk klasifikasi makanan
6
+ food_classifier = pipeline(task="image-classification", model="mhdiqbalpradipta/minang_food_classification")
7
+
8
+ def prediksi_makanan(gambar):
9
+ # Melakukan prediksi menggunakan model
10
+ hasil = food_classifier(images=gambar)[0]
11
+
12
+ # Menyimpan label dan skor
13
+ label_makanan = hasil['label']
14
+ skor = hasil['score']
15
+
16
+ return f"Makanan: {label_makanan}, Skor: {skor:.2f}"
17
+
18
+ device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
19
+ print(f"Menggunakan perangkat: {device}")
20
+
21
+ # Antarmuka Gradio
22
+ image_in = gr.inputs.Image(type='pil')
23
+ label_out = gr.outputs.Label()
24
+ contoh_gambar = ['ayam_goreng.jpg', 'ayam_pop.jpg', 'daging_rendang.jpg', 'dendeng_batokok.jpg', 'gulai_ikan.jpg', 'gulai_tambusu.jpg', 'gulai_tunjang', 'telur_balado.jpg', 'telur_dadar.jpg']
25
+
26
+ 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.", live=True)
27
+ intf.launch()