SakibRumu commited on
Update app.py
Browse files
app.py
CHANGED
|
@@ -55,7 +55,7 @@ def recognize_image(image):
|
|
| 55 |
#import pdb;pdb.set_trace()
|
| 56 |
pred,idx,probs=model.predict(image)
|
| 57 |
print(pred)
|
| 58 |
-
return dict(zip(
|
| 59 |
|
| 60 |
image = gr.Image()
|
| 61 |
label = gr.Label()
|
|
|
|
| 55 |
#import pdb;pdb.set_trace()
|
| 56 |
pred,idx,probs=model.predict(image)
|
| 57 |
print(pred)
|
| 58 |
+
return dict(zip(food_labels, map(float,probs)))
|
| 59 |
|
| 60 |
image = gr.Image()
|
| 61 |
label = gr.Label()
|