Update app.py

app.py

@@ -50,11 +50,8 @@ def generate_image(prompt, seed=24, width=DEFAULT_WIDTH, height=DEFAULT_HEIGHT,
             generator=generator
         )
     latency = f"Latency: {(time.time()-start_time):.2f} seconds"
-    image= img
-    img_byte_array = BytesIO()
-    image.save(img_byte_array, format="PNG", optimize=False)
-    img_byte_array.seek(0)
-    return img_byte_array, seed, latency
+    
+    return img, seed, latency
 
 # Example prompts
 examples = [