Update app.py
Browse files
app.py
CHANGED
@@ -79,7 +79,7 @@ def bigscience(model_name, sl, tl, input_text):
|
|
79 |
def bloomz(model_name, sl, tl, input_text):
|
80 |
tokenizer = AutoTokenizer.from_pretrained(model_name)
|
81 |
model = AutoModelForCausalLM.from_pretrained(model_name)
|
82 |
-
inputs = tokenizer.encode(f"Translate to {tl}
|
83 |
outputs = model.generate(inputs)
|
84 |
translation = tokenizer.decode(outputs[0])
|
85 |
translation = translation.replace('<pad> ', '').replace('</s>', '')
|
|
|
79 |
def bloomz(model_name, sl, tl, input_text):
|
80 |
tokenizer = AutoTokenizer.from_pretrained(model_name)
|
81 |
model = AutoModelForCausalLM.from_pretrained(model_name)
|
82 |
+
inputs = tokenizer.encode(f"Translate from {sl} to {tl}: {input_text}. Translation:", return_tensors="pt")
|
83 |
outputs = model.generate(inputs)
|
84 |
translation = tokenizer.decode(outputs[0])
|
85 |
translation = translation.replace('<pad> ', '').replace('</s>', '')
|