Updated

app.py

@@ -7,11 +7,18 @@ import numpy as np
 detector = pipeline("object-detection", model="mdefrance/yolos-tiny-signature-detection")
 
 # Fix: Convert NumPy to PIL image
-def detect_signature(image: np.ndarray):
+def detect_signature(image):
     if image is None:
         return {"error": "❌ Please draw or upload a signature before submitting."}
 
-    pil_image = Image.fromarray(image.astype("uint8"), "RGB")
+    # Normalize input structure
+    if isinstance(image, dict):
+        image = image.get("image", None)
+
+    if image is None:
+        return {"error": "❌ No image data received."}
+
+    pil_image = Image.fromarray(np.array(image).astype("uint8"), "RGB")
     results = detector(pil_image)
 
     valid = [r for r in results if r["score"] > 0.3]
@@ -20,6 +27,7 @@ def detect_signature(image: np.ndarray):
 
     return {"message": "✅ Signature detected.", "detections": valid}
 
+
 # Interface
 gr.Interface(
     fn=detect_signature,