Spaces:
Paused
Paused
Update app.py
Browse files
app.py
CHANGED
|
@@ -19,7 +19,7 @@ from rag_engine import RAGEngine
|
|
| 19 |
from llm_client import LLMClient
|
| 20 |
|
| 21 |
# --------- Config ----------
|
| 22 |
-
REPO_ID = "
|
| 23 |
CKPT_NAME = "deberta_lstm_checkpoint.pt" # the .pt file name
|
| 24 |
MODEL_NAME = "microsoft/deberta-base" # base tokenizer/backbone
|
| 25 |
LABELS = ["benign", "phishing"] # adjust to your classes
|
|
@@ -345,13 +345,11 @@ deberta_interface = gr.Interface(
|
|
| 345 |
["http://paypaI-security-update.net/login"],
|
| 346 |
["Your package has been delivered successfully. Thank you for using our service."],
|
| 347 |
["https://github.com/user/repo"]
|
| 348 |
-
]
|
| 349 |
-
theme=gr.themes.Soft(),
|
| 350 |
-
css=css_style
|
| 351 |
)
|
| 352 |
|
| 353 |
# Tab 2: LLM + RAG
|
| 354 |
-
with gr.Blocks(
|
| 355 |
gr.Markdown("# 🤖 AI Assistant (RAG)")
|
| 356 |
gr.Markdown("""
|
| 357 |
**AI Assistant** uses **Qwen2.5-1.5B** + **LangChain** to explain *why* a message is suspicious.
|
|
|
|
| 19 |
from llm_client import LLMClient
|
| 20 |
|
| 21 |
# --------- Config ----------
|
| 22 |
+
REPO_ID = "khoa-done/phishing-detector" # HF repo that holds the checkpoint
|
| 23 |
CKPT_NAME = "deberta_lstm_checkpoint.pt" # the .pt file name
|
| 24 |
MODEL_NAME = "microsoft/deberta-base" # base tokenizer/backbone
|
| 25 |
LABELS = ["benign", "phishing"] # adjust to your classes
|
|
|
|
| 345 |
["http://paypaI-security-update.net/login"],
|
| 346 |
["Your package has been delivered successfully. Thank you for using our service."],
|
| 347 |
["https://github.com/user/repo"]
|
| 348 |
+
]
|
|
|
|
|
|
|
| 349 |
)
|
| 350 |
|
| 351 |
# Tab 2: LLM + RAG
|
| 352 |
+
with gr.Blocks() as rag_interface:
|
| 353 |
gr.Markdown("# 🤖 AI Assistant (RAG)")
|
| 354 |
gr.Markdown("""
|
| 355 |
**AI Assistant** uses **Qwen2.5-1.5B** + **LangChain** to explain *why* a message is suspicious.
|