Update app.py

app.py

@@ -43,9 +43,7 @@ def generate_image(period1, period2, interpolation_value):
     generated_image = generated_image[0][0].detach().cpu().numpy()
     generated_image = (generated_image * 255).astype(np.uint8)
     pil_img = PILImage.fromarray(generated_image)
-    img_byte_arr = io.BytesIO()
-    pil_img.save(img_byte_arr, format='PNG')
-    return img_byte_arr.getvalue()
+    return pil_img
 
 # Define Gradio interface
 iface = gr.Interface(