Update app.py

app.py

@@ -37,9 +37,9 @@ def api_call(image_base64_file):
     return res_image
 
 
-def predict(dict, prompt):
+def predict(inputs):
 
-    init_image = Image.fromarray(dict['background'][:, :, :3], 'RGB') #dict['background'].convert("RGB")#.resize((1024, 1024))
+    init_image = Image.fromarray(inputs[0], 'RGB') #dict['background'].convert("RGB")#.resize((1024, 1024))
     
     image_base64_file = convert_image_to_base64_string(init_image)