Update app.py

app.py

@@ -28,14 +28,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"
-    weights_path = "./Ai-Text-Detector/model_weights.pth"
     tokenizer = GPT2TokenizerFast.from_pretrained(model_path)
-    model = GPT2LMHeadModel.from_pretrained("gpt2",from_tf=True)
+    
+    # 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"
     model.load_state_dict(torch.load(weights_path, map_location=torch.device("cpu")))
-    model.eval()  # Set the model to evaluation mode
+    
+    # Set the model to evaluation mode
+    model.eval()
+    
     return model, tokenizer