grammarchecker / app.py
nlpguy's picture
fix
7a418f1
raw
history blame contribute delete
323 Bytes
from transformers import pipeline
import gradio as gr
pipe = pipeline("text-classification", model="agentlans/snowflake-arctic-xs-grammar-classifier")
def process(text):
predictions = pipe(text)
return predictions
demo = gr.Interface(
fn=process,
inputs=["text"],
outputs=["text"],
)
demo.launch()