Spaces:
Build error
Build error
set the number of suggestions to 1
Browse files
app.py
CHANGED
@@ -25,6 +25,7 @@ def feedback(text):
|
|
25 |
# tokenized_phrases = grammar_tokenizer([text], return_tensors='pt', padding=True)
|
26 |
# corrections = grammar_model.generate(**tokenized_phrases)
|
27 |
# corrections = grammar_tokenizer.batch_decode(corrections, skip_special_tokens=True)
|
|
|
28 |
batch = grammar_tokenizer([text],truncation=True,padding='max_length',max_length=64, return_tensors="pt")
|
29 |
corrections= grammar_model.generate(**batch,max_length=64,num_beams=2, num_return_sequences=num_return_sequences, temperature=1.5)
|
30 |
print("The corrections are: ", corrections)
|
|
|
25 |
# tokenized_phrases = grammar_tokenizer([text], return_tensors='pt', padding=True)
|
26 |
# corrections = grammar_model.generate(**tokenized_phrases)
|
27 |
# corrections = grammar_tokenizer.batch_decode(corrections, skip_special_tokens=True)
|
28 |
+
num_return_sequences=1
|
29 |
batch = grammar_tokenizer([text],truncation=True,padding='max_length',max_length=64, return_tensors="pt")
|
30 |
corrections= grammar_model.generate(**batch,max_length=64,num_beams=2, num_return_sequences=num_return_sequences, temperature=1.5)
|
31 |
print("The corrections are: ", corrections)
|