Jour commited on
Commit
2878ae6
1 Parent(s): 52b6652

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -32,7 +32,7 @@ def translate(input, output, text):
32
  response = requests.request("POST", API_URL, json=json_)
33
  output = response.json()[0]['generated_text']
34
 
35
- return output.replace(instruction, '', 1)
36
 
37
  demo = gr.Blocks()
38
 
 
32
  response = requests.request("POST", API_URL, json=json_)
33
  output = response.json()[0]['generated_text']
34
 
35
+ return output.replace(instruction, '', 1).split('<end>')[0]
36
 
37
  demo = gr.Blocks()
38