Update handler.py

handler.py

@@ -1,7 +1,7 @@
 from transformers import AutoTokenizer, AutoModelForSequenceClassification
 import torch
 
-class Handler:
+class EndpointHandler:
     def __init__(self):
         self.model = AutoModelForSequenceClassification.from_pretrained(".")
         self.tokenizer = AutoTokenizer.from_pretrained(".")
@@ -12,4 +12,4 @@ class Handler:
         with torch.no_grad():
             logits = self.model(**encoded).logits
         predicted_class = torch.argmax(logits, dim=1).item()
-        return {"label": self.labels[predicted_class], "score": 1.0}  # score simplifié
+        return {"label": self.labels[predicted_class], "score": 1.0}