Spaces:
Runtime error
Runtime error
Bug fix
Browse files
app.py
CHANGED
@@ -8,7 +8,7 @@ def classify_image(img):
|
|
8 |
results = dict(zip(list(learn.dls.vocab),probs))
|
9 |
best_results = sorted(results, key=results.get, reverse=True)[:5]
|
10 |
probabilities = {key: float(results[key]) for key in best_results}
|
11 |
-
return
|
12 |
|
13 |
image = gr.inputs.Image(shape=(224,224))
|
14 |
label = gr.outputs.Label()
|
|
|
8 |
results = dict(zip(list(learn.dls.vocab),probs))
|
9 |
best_results = sorted(results, key=results.get, reverse=True)[:5]
|
10 |
probabilities = {key: float(results[key]) for key in best_results}
|
11 |
+
return probabilities
|
12 |
|
13 |
image = gr.inputs.Image(shape=(224,224))
|
14 |
label = gr.outputs.Label()
|