Update app.py

app.py

@@ -7,6 +7,7 @@ import requests
 from io import BytesIO
 import io
 import base64
+from loadimg import load_img
 
 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}
@@ -37,11 +38,10 @@ def api_call(image_base64_file):
     return res_image
 
 
-def predict(inputs):
-
-    init_image = Image.fromarray(inputs[0], 'RGB') #dict['background'].convert("RGB")#.resize((1024, 1024))
-    
-    image_base64_file = convert_image_to_base64_string(init_image)
+def predict(img_input):
+    img = load_img(img_input
+    img = img.convert("RGB")  
+    image_base64_file = convert_image_to_base64_string(img)
     
     gen_img = api_call(image_base64_file)
     
@@ -115,7 +115,7 @@ with image_blocks as demo:
             image_out = gr.Image(label="Output", elem_id="output-img")
 
     # Button click will trigger the inpainting function (now with prompt included)
-    btn.click(fn=predict, inputs=[image], outputs=[image_out], api_name='run')
+    btn.click(fn=predict, inputs=image, outputs=[image_out], api_name='run')
 
     gr.HTML(
         """