Update app.py

app.py

@@ -1,20 +1,35 @@
 import gradio as gr
 from transformers import pipeline
 
+# demo = gr.load("xy4286/yang-grammer-check", src="models")
+# def greet(name):
+#     return "Hello " + name + "!"
+# demo = gr.Interface(fn=greet, inputs="text", outputs="text")
+# demo.launch()
 
 
-demo = gr.load("xy4286/yang-grammer-check", src="models")
 
+pipe = pipeline("text-classification", model="xy4286/yang-grammer-check")
 
 
+def predict(sentence):
+  result = pipe(sentence)
+  # print(result)
+  if result[0]['label'] == 'LABEL_1' and result[0]['score'] > 0.5:
+    return "Grammatically Correct"
+  elif result[0]['label'] == 'LABEL_0' and result[0]['score'] > 0.5:
+    return "Grammatically Incorrect"
+  return result[0]
 
-def greet(name):
-    return "Hello " + name + "!"
-
-demo = gr.Interface(fn=greet, inputs="text", outputs="text")
-
+demo = gr.Interface(
+    fn=predict,
+    inputs = gr.Textbox(label="Please input a sentence to check"),
+    outputs = gr.Label(label="check result")
+    
+)
 
 
+# if __name__ == "__main__":
 demo.launch()