sudthi commited on
Commit
0b98236
·
verified ·
1 Parent(s): cca207d

สร้าง app.py

Browse files
Files changed (1) hide show
  1. app.py +20 -0
app.py ADDED
@@ -0,0 +1,20 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import gradio as gr
2
+ from transformers import pipeline
3
+
4
+ pipeline = pipeline(task="image-classification",model="julien-c/hotdog-not-hotdog")
5
+
6
+ def predict(input_img):
7
+ prediction = pipeline(input_img)
8
+ return input_img, {p["lable"]: p["score"] for p in prediction}
9
+
10
+ gradio_app = gr.interface(
11
+ predict,
12
+ input=gr.image(label="Select hot dog candidate", source="['upload', 'webcam']", type="pil"),
13
+ outputs=[gr.image(label="Processed Image"), gr.label="Result",num_top_classes=2],
14
+ title="Hot dog? or Not",
15
+
16
+ )
17
+
18
+ if__name__=="__main__":
19
+ gradio_app,launch()
20
+