Загрузка модели обнаружения фишинговых сообщений

README.md

@@ -1,12 +1,12 @@
 ---
-title: Phishing Or Not Demo Rubert
-emoji: 🐠
-colorFrom: green
-colorTo: red
+title: Классификация фишинговых сообщений
+colorFrom: blue
+colorTO: yellow
 sdk: gradio
-sdk_version: 5.25.0
+sdk_version: 5.9.1
 app_file: app.py
 pinned: false
+licence: apache-2.0
 ---
 
-Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
+Является ли сообщение фишинговым?

app.py

@@ -0,0 +1,33 @@
+import torch
+import gradio as gr
+from typing import Dict
+from transformers import pipeline
+
+def phishing_not_phishing_classifier(text:str) -> Dict[str, float]:
+  phishing_not_phishing_classifier_pipeline = pipeline(task="text-classification",
+                                                       model="twoweeksgetheart/learn_hf_phishing_not_phishing_text_classifier_rubert",
+                                                       batch_size=32,
+                                                       device="cuda" if torch.cuda.is_available() else "cpu",
+                                                       top_k=None)
+  outputs = phishing_not_phishing_classifier_pipeline(text)[0]
+
+  outpid_dict = {}
+  for item in outputs:
+    outpid_dict[item["label"]] = item["score"]
+
+  return outpid_dict
+
+description = """
+Является ли сообщение фишинговым?
+"""
+
+demo = gr.Interface(
+    fn=phishing_not_phishing_classifier,
+    inputs="text",
+    outputs=gr.Label(num_top_classes=2),
+    title="Классификация фишинговых сообщений",
+    description=description,
+)
+
+if __name__ == "__main__":
+  demo.launch()

requirements.txt

@@ -0,0 +1,3 @@
+gradio
+torch
+transformers