Update app.py

app.py

@@ -1,25 +1,19 @@
 
-from gramformer import Gramformer
-import torch
+from transformers import pipeline
 import gradio as gr
 
-def set_seed(seed):
-  torch.manual_seed(seed)
-  if torch.cuda.is_available():
-    torch.cuda.manual_seed_all(seed)
 
-set_seed(1212)
-
-gf = Gramformer(models = 1, use_gpu=False) # 1=corrector, 2=detector
+corrector = pipeline(
+              'text2text-generation',
+              'pszemraj/flan-t5-large-grammar-synthesis',
+              )
 
 def correct(text):
     sentences = text.split(".")
     output = ""
     for sentence in sentences:
         sentence += "."
-        correctedSet = gf.correct(sentence , max_candidates=1)
-        for corrected in correctedSet:
-            output += corrected
+        output += corrector(sentence)
     return output
 
 iface = gr.Interface(
@@ -31,3 +25,6 @@ iface = gr.Interface(
 
 # Launch the Gradio interface
 iface.launch(share=True)
+
+
+