Update app.py

app.py

@@ -11,7 +11,13 @@ import base64
 hf_token = os.environ.get("HF_TOKEN_API_DEMO") # we get it from a secret env variable, such that it's private
 auth_headers = {"api_token": hf_token}
 
-
+def convert_image_to_base64_string(img):
+    buffer = io.BytesIO()
+    img.save(buffer, format="PNG")  # You can choose the format (e.g., "JPEG", "PNG")
+    # Encode the buffer in base64
+    image_base64_string = base64.b64encode(buffer.getvalue()).decode('utf-8')
+    return f",{image_base64_string}"
+    
 def download_image(url):
     response = requests.get(url)
     img_bytes = BytesIO(response.content)
@@ -35,7 +41,7 @@ def predict(dict, prompt):
 
     init_image = Image.fromarray(dict['background'][:, :, :3], 'RGB') #dict['background'].convert("RGB")#.resize((1024, 1024))
     
-    image_base64_file = convert_mask_image_to_base64_string(init_image)
+    image_base64_file = convert_image_to_base64_string(init_image)
     
     gen_img = api_call(image_base64_file, mask_base64_file, prompt)