File size: 346 Bytes
dca9cd6
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
import gradio as gr
from transformers import pipeline

def correct_text(text):
    model_id = "grammarly/coedit-large"
    pipe = pipeline("text2text-generation", model=model_id)
    corrected = pipe(text)[0]['generated_text']
    return corrected

interface = gr.Interface(fn=correct_text, inputs="text_area", outputs="text")
interface.launch()