Adhir commited on
Commit
5f3c8d1
1 Parent(s): ebf855d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -7,7 +7,7 @@ pipeline = pipeline(task="image-classification", model="julien-c/hotdog-not-hotd
7
 
8
  def predict(image):
9
  predictions = pipeline(image)
10
- return [p["label"]: p["score"] for p in predictions]
11
 
12
  gr.interface(
13
  predict,
 
7
 
8
  def predict(image):
9
  predictions = pipeline(image)
10
+ return {p["label"]: p["score"] for p in predictions}
11
 
12
  gr.interface(
13
  predict,