nugentc commited on
Commit
55f3dd8
·
1 Parent(s): c669d92

set the number of suggestions to 1

Browse files
Files changed (1) hide show
  1. app.py +1 -0
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)