dungeon29 commited on
Commit
ca70721
·
verified ·
1 Parent(s): c020c14

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -35,8 +35,8 @@ from rag_engine import RAGEngine
35
  from llm_client import LLMClient
36
 
37
  # --------- Config ----------
38
- REPO_ID = "dungeon29/phishing-deberta-lstm" # HF repo that holds the checkpoint
39
- CKPT_NAME = "pytorch_model.bin" # the .pt file name
40
  MODEL_NAME = "microsoft/deberta-base" # base tokenizer/backbone
41
  LABELS = ["benign", "phishing"] # adjust to your classes
42
 
 
35
  from llm_client import LLMClient
36
 
37
  # --------- Config ----------
38
+ REPO_ID = "dungeon29/deberta-lstm-detect-phishing"
39
+ CKPT_NAME = "pytorch_model.bin"
40
  MODEL_NAME = "microsoft/deberta-base" # base tokenizer/backbone
41
  LABELS = ["benign", "phishing"] # adjust to your classes
42