Update handler.py

handler.py

@@ -1,9 +1,9 @@
-model_name_or_path="alfaxadeyembe/gemma2-27b-swahili-it"
+model_name="alfaxadeyembe/gemma2-27b-swahili-it"
 from transformers import AutoModelForSequenceClassification, AutoTokenizer
 import torch
 
 class EndpointHandler:
-    def __init__(self, model_name_or_path):
+    def __init__(self, model_name):
         self.tokenizer = AutoTokenizer.from_pretrained(model_name_or_path)
         self.model = AutoModelForSequenceClassification.from_pretrained(model_name_or_path)
         self.model.eval()  # Set the model to evaluation mode