salmanmapkar commited on
Commit
7ae24a2
·
1 Parent(s): 0d03722

Update app.py

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