Spaces:
Runtime error
Runtime error
Commit
·
7ae24a2
1
Parent(s):
0d03722
Update app.py
Browse files
app.py
CHANGED
@@ -18,7 +18,7 @@ model = T5ForConditionalGeneration.from_pretrained('deep-learning-analytics/Gram
|
|
18 |
|
19 |
def correct_grammar(input_text,num_return_sequences=num_return_sequences):
|
20 |
batch = tokenizer([input_text],truncation=True,padding='max_length',max_length=64, return_tensors="pt").to(torch_device)
|
21 |
-
results = model.generate(**batch,
|
22 |
#answer = tokenizer.batch_decode(results[0], skip_special_tokens=True)
|
23 |
return results
|
24 |
|
|
|
18 |
|
19 |
def correct_grammar(input_text,num_return_sequences=num_return_sequences):
|
20 |
batch = tokenizer([input_text],truncation=True,padding='max_length',max_length=64, return_tensors="pt").to(torch_device)
|
21 |
+
results = model.generate(**batch,num_beams=2, num_return_sequences=num_return_sequences, temperature=1.5)
|
22 |
#answer = tokenizer.batch_decode(results[0], skip_special_tokens=True)
|
23 |
return results
|
24 |
|