mamiksik commited on
Commit
233785c
1 Parent(s): 5bc6a2b
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -68,7 +68,7 @@ def predict(patch, max_length, min_length, num_beams, prediction_count):
68
  )
69
 
70
  result = tokenizer.batch_decode(outputs, skip_special_tokens=True)
71
- return token_count, '\n'.join(accumulator), {k: 0 for k in result}
72
 
73
 
74
  iface = gr.Interface(fn=predict, inputs=[
 
68
  )
69
 
70
  result = tokenizer.batch_decode(outputs, skip_special_tokens=True)
71
+ return token_count, input_text, {k: 0 for k in result}
72
 
73
 
74
  iface = gr.Interface(fn=predict, inputs=[