Spaces:
Runtime error
Runtime error
Simon Salmon
commited on
Commit
·
c40d011
1
Parent(s):
37dea39
Update app.py
Browse files
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([
|
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
|