Spaces:
Running
Running
Update app.py
Browse files
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
|
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):
|