Update app.py

app.py

@@ -1,6 +1,8 @@
 import gradio as gr
 import torch
 from diffusers import DiffusionPipeline
+import base64
+from io import BytesIO
 
 def load_amused_model():
     return DiffusionPipeline.from_pretrained("amused/amused-256")
@@ -21,12 +23,16 @@ def inference(prompt):
     if error:
         print(f"Error generating image: {error}")  # Debugging statement
         return "Error: " + error
-    return image
+    
+    buffered = BytesIO()
+    image.save(buffered, format="PNG")
+    img_str = base64.b64encode(buffered.getvalue()).decode("utf-8")
+    return img_str
 
 gradio_interface = gr.Interface(
     fn=inference,
     inputs="text",
-    outputs="image"
+    outputs="text"  # Change output to text to return base64 string
 )
 
 if __name__ == "__main__":