accplan commited on
Commit
8975744
β€’
1 Parent(s): 7b7a295
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -2,7 +2,7 @@ import gradio as gr
2
  from happytransformer import HappyTextToText, TTSettings
3
 
4
  happy_tt = HappyTextToText("T5", "vennify/t5-base-grammar-correction")
5
- args = TTSettings(num_beams=5, min_length=1)
6
 
7
  def fix(bad):
8
  result = happy_tt.generate_text("grammar: " + bad, args=args)
 
2
  from happytransformer import HappyTextToText, TTSettings
3
 
4
  happy_tt = HappyTextToText("T5", "vennify/t5-base-grammar-correction")
5
+ args = TTSettings(num_beams=5, min_length=1, max_length=999999)
6
 
7
  def fix(bad):
8
  result = happy_tt.generate_text("grammar: " + bad, args=args)