Update app.py

app.py

@@ -1,14 +1,14 @@
 import json
+import math
 import torch
 import torch.nn as nn
 import torch.nn.functional as F
-from fastapi import FastAPI, HTTPException
+from fastapi import FastAPI
 from pydantic import BaseModel
 from typing import List
-import math
 
 # ----------------------------
-# 1. Model definition (same as training)
+# 1. Model definition (must match training)
 # ----------------------------
 class PositionalEncoding(nn.Module):
     def __init__(self, d_model: int, max_len: int = 512, dropout: float = 0.1):
@@ -54,7 +54,7 @@ class PhishingTransformer(nn.Module):
         return self.head(x)
 
 # ----------------------------
-# 2. Tokenizer (same as training)
+# 2. Character tokenizer
 # ----------------------------
 class CharTokenizer:
     def __init__(self, max_len: int, char2id: dict):
@@ -73,9 +73,11 @@ class CharTokenizer:
 # ----------------------------
 with open("config.json", "r") as f:
     config = json.load(f)
+
 with open("tokenizer.json", "r") as f:
     tok_data = json.load(f)
 tokenizer = CharTokenizer(max_len=tok_data["max_len"], char2id=tok_data["char2id"])
+
 threshold = float(open("threshold.txt").read().strip())
 
 model = PhishingTransformer(config)