Upload handler.py with huggingface_hub

handler.py

@@ -1,22 +1,19 @@
 
-from typing import Dict, List, Any
 import base64
 import io
 from PIL import Image, ImageDraw
-import numpy as np
-import torch
 import json
 
 class EndpointHandler:
     def __init__(self, path=""):
-        self.initialized = False
         self.path = path
+        self.initialized = False
         
     def __call__(self, data):
         """Handle a request to the model."""
         if not self.initialized:
             self.initialize()
-            
+        
         if data is None:
             return None
         
@@ -30,14 +27,19 @@ class EndpointHandler:
         
     def preprocess(self, request):
         """Process the input request."""
-        inputs = request.pop("inputs", {})
-        if isinstance(inputs, str):
+        if isinstance(request, str):
             # Single prompt
-            prompt = inputs
+            prompt = request
             payload = {"prompt": prompt}
-        else:
+        elif isinstance(request, dict):
             # Full payload
-            payload = inputs
+            payload = request
+        else:
+            # Try to parse as JSON
+            try:
+                payload = json.loads(request)
+            except:
+                payload = {"prompt": str(request)}
             
         return payload