Update app.py

app.py

@@ -29,22 +29,21 @@ def verify_token(auth: str):
 # Function to load model and tokenizer
 
 def load_model():
-    # Load tokenizer from the specified path
     model_path = "./Ai-Text-Detector/model"
-    tokenizer = GPT2TokenizerFast.from_pretrained(model_path)
-    
-    # Load the GPT-2 model (without loading weights initially)
-    model = GPT2LMHeadModel.from_pretrained("gpt2", config=model_path)
-    
-    # Load the custom weights from the local .pth file
     weights_path = "./Ai-Text-Detector/model_weights.pth"
+
+    # Load tokenizer and config from your custom model path
+    tokenizer = GPT2TokenizerFast.from_pretrained(model_path)
+    config = GPT2Config.from_pretrained(model_path)
+
+    # Initialize model from config (don't load any weights from Hugging Face)
+    model = GPT2LMHeadModel(config)
+
+    # Load your saved PyTorch weights
     model.load_state_dict(torch.load(weights_path, map_location=torch.device("cpu")))
-    
-    # Set the model to evaluation mode
-    model.eval()
-    
-    return model, tokenizer
 
+    model.eval()  # Set to evaluation mode
+    return model, tokenizer
 
 @asynccontextmanager
 async def lifespan(app: FastAPI):