Spaces:
Running
Running
updated app
Browse files
app.py
CHANGED
@@ -30,7 +30,7 @@ model = load_learner(model_path)
|
|
30 |
|
31 |
def recognize_image(image):
|
32 |
pred, idx, probs = model.predict(image)
|
33 |
-
return dict(zip(
|
34 |
|
35 |
image = gr.inputs.Image(shape=(192, 192))
|
36 |
label = gr.outputs.Label()
|
|
|
30 |
|
31 |
def recognize_image(image):
|
32 |
pred, idx, probs = model.predict(image)
|
33 |
+
return dict(zip(chicken_breeds, map(float, probs)))
|
34 |
|
35 |
image = gr.inputs.Image(shape=(192, 192))
|
36 |
label = gr.outputs.Label()
|