Spaces:
Runtime error
Runtime error
Commit
·
0b7dac5
1
Parent(s):
7ae24a2
Update app.py
Browse files
app.py
CHANGED
@@ -17,7 +17,7 @@ tokenizer = T5Tokenizer.from_pretrained('deep-learning-analytics/GrammarCorrecto
|
|
17 |
model = T5ForConditionalGeneration.from_pretrained('deep-learning-analytics/GrammarCorrector').to(torch_device)
|
18 |
|
19 |
def correct_grammar(input_text,num_return_sequences=num_return_sequences):
|
20 |
-
batch = tokenizer([input_text],truncation=True,
|
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
|
|
|
17 |
model = T5ForConditionalGeneration.from_pretrained('deep-learning-analytics/GrammarCorrector').to(torch_device)
|
18 |
|
19 |
def correct_grammar(input_text,num_return_sequences=num_return_sequences):
|
20 |
+
batch = tokenizer([input_text],truncation=True, 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
|