Update app.py

app.py

@@ -52,13 +52,7 @@ def generate(prompt: str,
                 generator=generator,
                 output_type='pil').images[0]
     
-    # Convert PIL image to a byte stream
-    buffered = io.BytesIO()
-    image.save(buffered, format="PNG")
-    
-    # Encode to base64
-    img_str = base64.b64encode(buffered.getvalue()).decode()
-    return img_str
+    return image
 
 with gr.Blocks() as demo:
     gr.HTML("""
@@ -80,7 +74,7 @@ with gr.Blocks() as demo:
         placeholder='Enter your prompt',
         container=False,
     )
-    result = gr.Text(label='Result', show_label=False, visible=True)
+    result = gr.Image(label='Result', show_label=False)
     negative_prompt = gr.Text(
         label='Negative prompt',
         max_lines=1,