KaizeShi commited on
Commit
f5b48c2
·
1 Parent(s): f456e05

Add application file

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -118,7 +118,7 @@ def evaluate(
118
  )
119
  s = generation_output.sequences[0]
120
  output = tokenizer.decode(s)
121
- return output.split("### Response:")[1].strip()
122
 
123
 
124
  g = gr.Interface(
 
118
  )
119
  s = generation_output.sequences[0]
120
  output = tokenizer.decode(s)
121
+ return output.split("### Response:")[1].strip().split("</s>")[0]
122
 
123
 
124
  g = gr.Interface(