Update handler.py

handler.py

@@ -7,10 +7,10 @@ from peft import PeftModel
 
 # Load model and tokenizer
 model_name = "munzirmuneer/phishing_url_gemma_pytorch" # Replace with your specific model
-#model_name2 = "google/gemma-2b"
-tokenizer = AutoTokenizer.from_pretrained(model_name, use_auth_token=True)
-model = AutoModelForSequenceClassification.from_pretrained(model_name, use_auth_token=True)
-model = PeftModel.from_pretrained(model, model_name, use_auth_token=True)
+model_name2 = "google/gemma-2b"
+tokenizer = AutoTokenizer.from_pretrained(model_name2)
+model = AutoModelForSequenceClassification.from_pretrained(model_name)
+model = PeftModel.from_pretrained(model, model_name)
 
 def predict(input_text):
     # Tokenize input