Simon Salmon commited on
Commit
c40d011
·
1 Parent(s): 37dea39

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -15,7 +15,7 @@ device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
15
  model = model.to(device)
16
 
17
  def get_response(text):
18
- batch = tokenizer([input_text],truncation=True,padding='longest',max_length=60, return_tensors="pt").to(device)
19
  translated = model.generate(**batch,max_length=60, do_sample=True, num_return_sequences=10, temperature=1.5)
20
  tgt_text = tokenizer.batch_decode(translated, skip_special_tokens=True)
21
  return tgt_text
 
15
  model = model.to(device)
16
 
17
  def get_response(text):
18
+ batch = tokenizer([text],truncation=True,padding='longest',max_length=60, return_tensors="pt").to(device)
19
  translated = model.generate(**batch,max_length=60, do_sample=True, num_return_sequences=10, temperature=1.5)
20
  tgt_text = tokenizer.batch_decode(translated, skip_special_tokens=True)
21
  return tgt_text