Update app.py

app.py

@@ -1,16 +1,12 @@
 import torch
 import torch.nn.functional as F
-from transformers import AutoTokenizer, AutoModelForSequenceClassification
+from transformers import AutoTokenizer
 from huggingface_hub import hf_hub_download
 import gradio as gr
 import requests
 import re
-from urllib.parse import urlparse
-from bs4 import BeautifulSoup
 import time
-import os
 import sys
-import asyncio
 import logging
 import urllib3  # Import urllib3 to handle warnings
 
@@ -39,7 +35,7 @@ from rag_engine import RAGEngine
 from llm_client import LLMClient
 
 # --------- Config ----------
-REPO_ID = "dungeon29/deberta-lstm-detect-phishing"       
+REPO_ID = "dungeon29/phishing-deberta-lstm"       # HF repo that holds the checkpoint
 CKPT_NAME = "pytorch_model.bin"               # the .pt file name
 MODEL_NAME = "microsoft/deberta-base"         # base tokenizer/backbone
 LABELS = ["benign", "phishing"]               # adjust to your classes