interface

app.py

@@ -1,4 +1,17 @@
-# Use a pipeline as a high-level helper
 from transformers import pipeline
+import gradio as gr
 
 pipe = pipeline("text-classification", model="agentlans/snowflake-arctic-xs-grammar-classifier")
+
+def process(text):
+    predictions = pipeline(text)
+    return predictions
+
+demo = gr.Interface(
+    fn=process,
+    inputs=["text"],
+    outputs=["text"],
+)
+
+demo.launch()
+