devtitan commited on
Commit
8fb66b4
·
verified ·
1 Parent(s): 8c23cb5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -10,7 +10,7 @@ app = Flask(__name__)
10
 
11
  # --- Load Tokenizer and Model (Using your provided code) ---
12
  # Load them globally when the app starts
13
- model_name = "amoshdahal/email-phishing-distilbert-finetuned"
14
  try:
15
  tokenizer = AutoTokenizer.from_pretrained(model_name)
16
  model = AutoModelForSequenceClassification.from_pretrained(model_name)
 
10
 
11
  # --- Load Tokenizer and Model (Using your provided code) ---
12
  # Load them globally when the app starts
13
+ model_name = "CrabInHoney/urlbert-tiny-v3-malicious-url-classifier"
14
  try:
15
  tokenizer = AutoTokenizer.from_pretrained(model_name)
16
  model = AutoModelForSequenceClassification.from_pretrained(model_name)