No image format

app.py

@@ -1,10 +1,10 @@
-import torch
-import io
 import spaces
 import gradio as gr
 import numpy as np
 from diffusers import AutoPipelineForText2Image
+import torch
 from huggingface_hub import snapshot_download
+from io import BytesIO
 from PIL import Image
 
 SPACE_USERNAME = 'KR_4dmin'
@@ -34,12 +34,7 @@ def infer(prompt, height, width, guidance_scale, num_inference_steps):
         # generator=generator
     ).images[0]
 
-    # Convert the image to JPG format
-    buffered = io.BytesIO()
-    image.save(buffered, format="JPEG")
-    jpeg_image = buffered.getvalue()
-
-    return jpeg_image
+    return image
 
 
 examples = [