Save as png instead of jpg

app.py

@@ -585,7 +585,7 @@ def generate_image(input_image_path, prompt):
 
     resized_input_img = resize_to_square(input_image_path, 512)
 
-    resized_input_img_path = resized_input_img.save('input_img.jpg')
+    resized_input_img_path = resized_input_img.save('input_img.png')
 
     ai_gen_image = generate_ai_bg(resized_input_img_path, prompt)