Modified image encoding

app.py

@@ -342,9 +342,9 @@ def upload_file(img_file):
 
 def encode_image(image):
     buffer = BytesIO()
-    image.save(buffer, format="PNG")  # You can change to JPEG if needed
+    image.save(buffer, format="PNG")
     encoded_image = base64.b64encode(buffer.getvalue()).decode("utf-8")
-    return encoded_image
+    return f"data:image/png;base64,{encoded_image}"
 
 def generate_ai_bg(input_img, prompt):
     hf_input_img = encode_image(input_img)