Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
|
@@ -113,7 +113,7 @@ def eurollm_instruct(model_name, sl, tl, input_text):
|
|
| 113 |
text = f'<|im_start|>system\n<|im_end|>\n<|im_start|>user\nTranslate the following {sl} source text to {tl}:\n{sl}: {input_text} \n{tl}: <|im_end|>\n<|im_start|>assistant\n'
|
| 114 |
inputs = tokenizer(text, return_tensors="pt")
|
| 115 |
outputs = model.generate(**inputs, max_new_tokens=256)
|
| 116 |
-
output = tokenizer.decode(outputs[0], skip_special_tokens=True)
|
| 117 |
if f'{tl}:' in output:
|
| 118 |
output = output.rsplit(f'{tl}:')[-1].strip()
|
| 119 |
return output
|
|
|
|
| 113 |
text = f'<|im_start|>system\n<|im_end|>\n<|im_start|>user\nTranslate the following {sl} source text to {tl}:\n{sl}: {input_text} \n{tl}: <|im_end|>\n<|im_start|>assistant\n'
|
| 114 |
inputs = tokenizer(text, return_tensors="pt")
|
| 115 |
outputs = model.generate(**inputs, max_new_tokens=256)
|
| 116 |
+
output = tokenizer.decode(outputs[0], skip_special_tokens=True)
|
| 117 |
if f'{tl}:' in output:
|
| 118 |
output = output.rsplit(f'{tl}:')[-1].strip()
|
| 119 |
return output
|