Update app.py

app.py

@@ -1,7 +1,6 @@
 import gradio as gr
 import torch
 from diffusers import AutoPipelineForText2Image
-import base64
 from io import BytesIO
 from generate_propmts import generate_prompt
 from concurrent.futures import ThreadPoolExecutor
@@ -13,11 +12,11 @@ model = AutoPipelineForText2Image.from_pretrained("stabilityai/sdxl-turbo")
 def generate_image(text, sentence_mapping, character_dict, selected_style):
     try:
         prompt, _ = generate_prompt(text, sentence_mapping, character_dict, selected_style)
+        print(f"Generated prompt: {prompt}")
         image = model(prompt=prompt, num_inference_steps=1, guidance_scale=0.0).images[0]
         buffered = BytesIO()
-        img_str = base64.b64encode(buffered.getvalue()).decode("utf-8")
-        if isinstance(result, img_str):
-            image_bytes = base64.b64decode(result)
+        image.save(buffered, format="JPEG")
+        image_bytes = buffered.getvalue()
         return image_bytes
     except Exception as e:
         print(f"Error generating image: {e}")