Convert webp in jpeg

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'
@@ -35,11 +35,11 @@ def infer(prompt, height, width, guidance_scale, num_inference_steps):
     ).images[0]
 
     # Convert the image to JPG format
-    img_byte_arr = BytesIO()
-    image.save(img_byte_arr, format='JPEG')
-    image_jpg = img_byte_arr.getvalue()
+    buffered = io.BytesIO()
+    image.save(buffered, format="JPEG")
+    jpeg_image = buffered.getvalue()
 
-    return image_jpg
+    return jpeg_image
 
 
 examples = [