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 = "dungeon29/
|
| 23 |
CKPT_NAME = "pytorch_model.bin" # the .pt file name
|
| 24 |
MODEL_NAME = "microsoft/deberta-base" # base tokenizer/backbone
|
| 25 |
LABELS = ["benign", "phishing"] # adjust to your classes
|
|
|
|
| 19 |
from llm_client import LLMClient
|
| 20 |
|
| 21 |
# --------- Config ----------
|
| 22 |
+
REPO_ID = "dungeon29/deberta-lstm-detect-phishing" # HF repo that holds the checkpoint
|
| 23 |
CKPT_NAME = "pytorch_model.bin" # the .pt file name
|
| 24 |
MODEL_NAME = "microsoft/deberta-base" # base tokenizer/backbone
|
| 25 |
LABELS = ["benign", "phishing"] # adjust to your classes
|