Timing0311 commited on
Commit
7994488
1 Parent(s): d655623

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -18,7 +18,7 @@ def translation_job(data):
18
  input = job_map[job] + text
19
  print(input)
20
  response = trans_tokenizer.batch_decode(input, max_length=100, num_beams=4)
21
- return response
22
 
23
 
24
 
 
18
  input = job_map[job] + text
19
  print(input)
20
  response = trans_tokenizer.batch_decode(input, max_length=100, num_beams=4)
21
+ return response[0]['generated_text']
22
 
23
 
24