Bert again

app.py

@@ -0,0 +1,30 @@
+import gradio as gr
+
+from transformers import BertConfig, BertModel
+
+config = BertConfig()
+model = BertModel(config)
+
+model = BertModel.from_pretrained("EricJedha/spam_finetuned_bert_overf")
+
+from transformers import AutoTokenizer
+from transformers import pipeline
+
+
+
+tokenizer = AutoTokenizer.from_pretrained(model)
+def check_spam(text):
+    classifier = pipeline("text-classification", model="EricJedha/spam_finetuned_bert_overf")
+    return classifier(text)[0]['label']
+
+with gr.Blocks() as demo:
+    with gr.Row():
+        with gr.Column():
+            sms = gr.Textbox(label = "SMS à checker")
+            spam_check_btn = gr.Button(label="Check")
+    with gr.Column():
+            result = gr.Textbox(label="Resultat de l'Analyse")
+
+spam_check_btn.click(check_spam, input = sms, outputs = result)
+
+demo.launch(share=True)