Create app.py

app.py

@@ -0,0 +1,22 @@
+from transformers import DistilBertTokenizer
+from transformers import DistilBertForSequenceClassification
+from transformers import pipeline
+import gradio as gr
+
+MODEL_PATH = "xiaopeiwu/hrw_v2"
+model = DistilBertForSequenceClassification.from_pretrained(MODEL_PATH)
+tokenizer = DistilBertTokenizer.from_pretrained("distilbert-base-uncased")
+
+demo = gr.Interface(
+    fn=clf_result,
+    title="Test High Risk Words model v2",
+    examples=["All the best lenders and rates for car loans in one AI powered marketplace", "Caregiver burnout can happen to your best employees."],
+    description="DistilBert for text classification model fine tuned on 70% of annotated RM production data combined with industry-specific webscrape data",
+    inputs=gr.Textbox(placeholder="Enter sentence here and press Submit", label="Sentence to check"),
+    outputs="textbox",
+)
+
+demo.launch(
+    share=True,
+    auth=("redmarkerml", "redmarkerml")
+    )