TiberiuCristianLeon commited on
Commit
7d0b5b1
·
verified ·
1 Parent(s): 03dd083

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -223,7 +223,7 @@ class Translators:
223
  outputs = model.generate(inputs)
224
  translation = tokenizer.decode(outputs[0])
225
  translation = translation.replace('<pad> ', '').replace('</s>', '')
226
- translation = translation.split('Translation:')[-1].strip() if 'Translation:' in output else translation.strip()
227
  return translation
228
 
229
  def nllb(self):
 
223
  outputs = model.generate(inputs)
224
  translation = tokenizer.decode(outputs[0])
225
  translation = translation.replace('<pad> ', '').replace('</s>', '')
226
+ translation = translation.split('Translation:')[-1].strip() if 'Translation:' in translation else translation.strip()
227
  return translation
228
 
229
  def nllb(self):