Update main.py

main.py

@@ -21,7 +21,7 @@ def generate():
     generator = torch.Generator('cuda').manual_seed(seed)
     image = pipe(text, num_inference_steps=25, generator=generator).images[0]
     img_io = BytesIO()
-    image.save(img_io)
+    image.save(img_io, 'PNG')
     img_io.seek(0)
     return send_file(img_io, mimetype='image/png')