dungeon29 commited on
Commit
13abf72
·
verified ·
1 Parent(s): 80c9366

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -5
app.py CHANGED
@@ -19,7 +19,7 @@ from rag_engine import RAGEngine
19
  from llm_client import LLMClient
20
 
21
  # --------- Config ----------
22
- REPO_ID = "dungeon29/phishing-deberta-lstm" # 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,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(theme=gr.themes.Soft(), css=css_style) as rag_interface:
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.