TiberiuCristianLeon commited on
Commit
79ff2d1
·
verified ·
1 Parent(s): 2861194

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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}, providing the most accurate translation: {input_text}.", return_tensors="pt")
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>', '')