futranbg commited on
Commit
a7dc181
1 Parent(s): 22ffa1e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -41,7 +41,7 @@ def translation(source, target, text):
41
  chunks = text.splitlines(True)
42
  output = ""
43
  for chunk in chunks:
44
- stchunk = translator(chunk, max_length=400, num_beams=5)
45
  output += stchunk[0]['translation_text']+"\n"
46
 
47
  end_time = time.time()
 
41
  chunks = text.splitlines(True)
42
  output = ""
43
  for chunk in chunks:
44
+ stchunk = translator(chunk+"<mask>", max_length=400, num_beams=5)
45
  output += stchunk[0]['translation_text']+"\n"
46
 
47
  end_time = time.time()