Update app.py

app.py

@@ -5,6 +5,7 @@ from PIL import Image
 from io import BytesIO
 from tqdm import tqdm
 import time
+import cairosvg
 
 # Defining the repository information and the trigger word
 repo = "artificialguybr/LineAniRedmond-LinearMangaSDXL-V2"
@@ -22,6 +23,7 @@ constraints = (
     "NEVER use anything besides a single black line of varying widths as necessary to achieve a result of the user's request."
 )
 
+
 def generate_image(prompt):
     print("Generating image with prompt:", prompt)
     api_url = f"https://api-inference.huggingface.co/models/{repo}"
@@ -51,7 +53,10 @@ def generate_image(prompt):
         print("API response status code:", response.status_code)
         if response.status_code == 200:
             print("Image generation successful!")
-            return Image.open(BytesIO(response.content))
+            # Convert the response content to an SVG with only black values
+            image = Image.open(BytesIO(response.content)).convert("L")  # Convert to grayscale
+            svg_data = cairosvg.svg2svg(bytestring=image.tobytes(), output_width=image.width, output_height=image.height)
+            return svg_data
         elif response.status_code == 503:
             time.sleep(1)
             pbar.update(1)
@@ -65,7 +70,7 @@ def generate_image(prompt):
 iface = gr.Interface(
     fn=generate_image,
     inputs=gr.Textbox(lines=2, placeholder="Describe the subject here..."),
-    outputs="image",
+    outputs="text",  # Changed output to text for SVG data
     title="LineArt XL Image Generator",
     description=(
         "Powered by the generous GPU time from Redmond.AI, this LORA, fine-tuned on SD XL 1.0, excels at creating Lineart-themed images across a wide range of subjects. "