Update app.py

app.py

@@ -14,9 +14,10 @@ def generate_image(text, sentence_mapping, character_dict, selected_style):
         prompt, _ = generate_prompt(text, sentence_mapping, character_dict, selected_style)
         image = model(prompt=prompt, num_inference_steps=1, guidance_scale=0.0).images[0]
         buffered = BytesIO()
-        image.save(buffered, format="JPEG")
         img_str = base64.b64encode(buffered.getvalue()).decode("utf-8")
-        return img_str
+        if isinstance(result, img_str):
+            image_bytes = base64.b64decode(result)
+        return image_bytes
     except Exception as e:
         print(f"Error generating image: {e}")
         return None