Update app.py

app.py

@@ -1,3 +1,32 @@
 import gradio as gr
+from transformers import BertTokenizer, BertForSequenceClassification
+from transformers import pipeline
 
-gr.Interface.load("models/elftsdmr/malware-url-detect").launch()
+urlbert = BertForSequenceClassification.from_pretrained('elftsdmr/malware-url-detect',num_labels=2)
+tokenizer = BertTokenizer.from_pretrained('elftsdmr/malware-url-detect')
+
+url_classifier = pipeline("text-classification", model=urlbert, tokenizer=tokenizer)
+
+
+
+
+
+def text_to_url(text):
+    return url_classifier(text)[0]["label"]
+
+
+demo = gr.Blocks(theme=gr.themes.Soft())
+
+with demo:
+   
+    text = gr.Textbox()
+    label = gr.Label()
+
+  
+    b2 = gr.Button("Classify URL")
+
+   
+    b2.click(text_to_url, inputs=text, outputs=label)
+
+if __name__ == "__main__":
+    demo.launch(share=True)