add easy split for label and probability
Browse files
app.py
CHANGED
|
@@ -35,6 +35,6 @@ def predict(image):
|
|
| 35 |
top_prob, top_idx = probs.topk(1, dim=1)
|
| 36 |
idx_to_class = {v:k for k,v in classes.items()}
|
| 37 |
label = idx_to_class[top_idx[0][0].item()]
|
| 38 |
-
return f"{label}
|
| 39 |
|
| 40 |
gr.Interface(fn=predict, inputs=gr.Image(type="pil"), outputs="text").launch()
|
|
|
|
| 35 |
top_prob, top_idx = probs.topk(1, dim=1)
|
| 36 |
idx_to_class = {v:k for k,v in classes.items()}
|
| 37 |
label = idx_to_class[top_idx[0][0].item()]
|
| 38 |
+
return f"{label}|({top_prob[0][0].item()})"
|
| 39 |
|
| 40 |
gr.Interface(fn=predict, inputs=gr.Image(type="pil"), outputs="text").launch()
|