Upload handler.py with huggingface_hub

handler.py

@@ -7,11 +7,24 @@ import numpy as np
 import torch
 import json
 
-class VectorGraphicsHandler:
-    def __init__(self):
+class EndpointHandler:
+    def __init__(self, path=""):
         self.initialized = False
+        self.path = path
         
-    def initialize(self, context):
+    def __call__(self, data):
+        """Handle a request to the model."""
+        if not self.initialized:
+            self.initialize()
+            
+        if data is None:
+            return None
+        
+        inputs = self.preprocess(data)
+        outputs = self.inference(inputs)
+        return self.postprocess(outputs)
+        
+    def initialize(self):
         """Initialize the handler."""
         self.initialized = True
         
@@ -70,17 +83,3 @@ class VectorGraphicsHandler:
     def postprocess(self, inference_output):
         """Return the output as JSON."""
         return inference_output
-
-_service = VectorGraphicsHandler()
-
-def handle(data, context):
-    """Handle a request to the model."""
-    if not _service.initialized:
-        _service.initialize(context)
-        
-    if data is None:
-        return None
-    
-    inputs = _service.preprocess(data)
-    outputs = _service.inference(inputs)
-    return _service.postprocess(outputs)