TiberiuCristianLeon commited on
Commit
1800b84
·
verified ·
1 Parent(s): b1a4240

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -70,7 +70,7 @@ def teuken(model_name, sl, tl, input_text):
70
  def bigscience(model_name, sl, tl, input_text):
71
  tokenizer = AutoTokenizer.from_pretrained(model_name)
72
  model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
73
- inputs = tokenizer.encode(f"Translate to {tl}: {input_text}", return_tensors="pt")
74
  outputs = model.generate(inputs)
75
  translation = tokenizer.decode(outputs[0])
76
  translation = translation.replace('<pad> ', '').replace('</s>', '')
@@ -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}: {input_text}", return_tensors="pt")
83
  outputs = model.generate(inputs)
84
  translation = tokenizer.decode(outputs[0])
85
  translation = translation.replace('<pad> ', '').replace('</s>', '')
 
70
  def bigscience(model_name, sl, tl, input_text):
71
  tokenizer = AutoTokenizer.from_pretrained(model_name)
72
  model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
73
+ inputs = tokenizer.encode(f"Translate to {tl}: {input_text}.", return_tensors="pt")
74
  outputs = model.generate(inputs)
75
  translation = tokenizer.decode(outputs[0])
76
  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 to {tl}: {input_text}.", return_tensors="pt")
83
  outputs = model.generate(inputs)
84
  translation = tokenizer.decode(outputs[0])
85
  translation = translation.replace('<pad> ', '').replace('</s>', '')