joey1895 commited on
Commit
93269e1
1 Parent(s): dac9e7a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -4,12 +4,13 @@ from transformers import pipeline
4
  pipeline = pipeline(task="image-classification", model="julien-c/hotdog-not-hotdog")
5
 
6
  def predict(image):
7
- predictions = pipeline(image)
8
- return {p["label"]: p["score"] for p in predictions}
9
 
10
  gr.Interface(
11
  predict,
12
  inputs=gr.inputs.Image(label="Upload hot dog candidate", type="filepath"),
13
  outputs=gr.outputs.Label(num_top_classes=2),
14
  title="Hot Dog? Or Not?",
 
15
  ).launch()
 
4
  pipeline = pipeline(task="image-classification", model="julien-c/hotdog-not-hotdog")
5
 
6
  def predict(image):
7
+ predictions = pipeline(image)
8
+ return {p["label"]: p["score"] for p in predictions}
9
 
10
  gr.Interface(
11
  predict,
12
  inputs=gr.inputs.Image(label="Upload hot dog candidate", type="filepath"),
13
  outputs=gr.outputs.Label(num_top_classes=2),
14
  title="Hot Dog? Or Not?",
15
+ allow_flagging="manual"
16
  ).launch()