Spaces:

leslyarun
/

grammar_correction

Runtime error

leslyarun commited on Oct 28, 2022

Commit

956953f

1 Parent(s): 9858aef

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -10,7 +10,7 @@ def check_grammar(sentence):
   text2text_generator = pipeline("text2text-generation", model=model, tokenizer=tokenizer)
   output = text2text_generator("grammar: " + sentence)
-  return output
 demo = gr.Interface(check_grammar, inputs=['text'],

   text2text_generator = pipeline("text2text-generation", model=model, tokenizer=tokenizer)
   output = text2text_generator("grammar: " + sentence)
+  return output[0]["generated_text"]
 demo = gr.Interface(check_grammar, inputs=['text'],