zlpnvrtnk commited on
Commit
beae59f
1 Parent(s): 56da6dc

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -0
app.py CHANGED
@@ -20,6 +20,9 @@ def captcha_solver(img):
20
  # # Greedy decoding
21
  pred = logits.softmax(-1)
22
  label, confidence = parseq.tokenizer.decode(pred)
 
 
 
23
  return label[0]
24
 
25
  demo = gr.Interface(fn=captcha_solver, inputs=gr.inputs.Image(type="pil"), outputs=gr.outputs.Textbox())
 
20
  # # Greedy decoding
21
  pred = logits.softmax(-1)
22
  label, confidence = parseq.tokenizer.decode(pred)
23
+ print(label)
24
+ print(confidence)
25
+
26
  return label[0]
27
 
28
  demo = gr.Interface(fn=captcha_solver, inputs=gr.inputs.Image(type="pil"), outputs=gr.outputs.Textbox())