Update handler.py

handler.py

@@ -1,7 +1,7 @@
 import torch
 from transformers import CLIPProcessor, CLIPModel
 
-class ModelHandler:
+class EndpointHandler:
     def __init__(self):
         # Load model and processor
         self.model = CLIPModel.from_pretrained("dazpye/clip-image")
@@ -23,7 +23,7 @@ class ModelHandler:
         # Convert output to readable format
         return {"predictions": inference_output}
 
-handler = ModelHandler()
+handler = EndpointHandler()
 
 def handle(request):
     inputs = request if isinstance(request, dict) else request.json()