Update handler to return PIL Images for Inference API compatibility

config.json

@@ -3,7 +3,7 @@
   "model_type": "diffsketcher",
   "task": "text-to-svg",
   "framework": "pytorch",
-  "pipeline_tag": "text-generation",
+  "pipeline_tag": "text-to-image",
   "library_name": "diffusers",
   "inference": {
     "parameters": {

handler.py

@@ -5,6 +5,9 @@ import torch
 import numpy as np
 from typing import Dict, Any, List
 import math
+from PIL import Image
+import cairosvg
+import io
 
 class EndpointHandler:
     def __init__(self, path=""):
@@ -90,12 +93,20 @@ class EndpointHandler:
             # Generate SVG content based on prompt
             svg_content = self._generate_sketch_svg(prompt, width, height, num_paths, guidance_scale)
             
-            # Return SVG as text for Inference API
-            return svg_content
+            # Convert SVG to PIL Image
+            try:
+                png_data = cairosvg.svg2png(bytestring=svg_content.encode('utf-8'))
+                image = Image.open(io.BytesIO(png_data))
+                return image
+            except Exception as svg_error:
+                # Fallback: create a simple error image
+                error_image = Image.new('RGB', (width, height), color='white')
+                return error_image
             
         except Exception as e:
-            # Return error SVG
-            return f'<svg width="224" height="224" xmlns="http://www.w3.org/2000/svg"><text x="10" y="20" fill="red">Error: {str(e)}</text></svg>'
+            # Return error image
+            error_image = Image.new('RGB', (224, 224), color='white')
+            return error_image
     
     def _generate_sketch_svg(self, prompt: str, width: int, height: int, num_paths: int, guidance_scale: float) -> str:
         """
@@ -147,7 +158,16 @@ class EndpointHandler:
         self._add_sketch_lines(paths, width, height, colors, min(20, num_paths // 5))
         
         svg_content = svg_header + '\n' + '\n'.join(paths) + '\n' + svg_footer
-        return svg_content
+        
+        # Convert SVG to PIL Image
+        try:
+            png_data = cairosvg.svg2png(bytestring=svg_content.encode('utf-8'))
+            image = Image.open(io.BytesIO(png_data))
+            return image
+        except Exception as e:
+            # Fallback: create a simple error image
+            error_image = Image.new('RGB', (width, height), color='white')
+            return error_image
     
     def _add_circular_elements(self, paths, width, height, colors, count):
         """Add circular elements to the SVG"""