codingmonster1234 commited on
Commit
7a0922e
·
verified ·
1 Parent(s): 16ca743

add easy split for label and probability

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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} ({top_prob[0][0].item()})"
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()