Update app.py

app.py

@@ -20,7 +20,7 @@ from rag_engine import RAGEngine
 from llm_client import LLMClient
 
 # --------- Config ----------
-REPO_ID = "khoa-done/phishing-detector"       
+REPO_ID = "dungeon29/deberta-lstm-detect-phishing"       
 CKPT_NAME = "pytorch_model.bin"               
 MODEL_NAME = "microsoft/deberta-base"         # base tokenizer/backbone
 LABELS = ["benign", "phishing"]               # adjust to your classes
@@ -41,7 +41,7 @@ model = DeBERTaLSTMClassifier()
 
 # Load weights
 try:
-    model.load_state_dict(checkpoint["model_state_dict"])
+    model.load_state_dict(checkpoint)
 except RuntimeError as e:
     if "attention" in str(e):
         # Old model without attention layer - initialize attention layer and load partial state