Handle result path

app.py

@@ -498,8 +498,8 @@ def generate_ai_bg(input_img, prompt):
             input_fg=handle_file(input_img),
             bg_source="None",
             prompt=prompt,
-            image_width=896,
-            image_height=1152,
+            image_width=1024,
+            image_height=1024,
             num_samples=1,
             seed=12345,
             steps=25,
@@ -514,7 +514,7 @@ def generate_ai_bg(input_img, prompt):
     )
     print(result)
 
-    relight_img_path = result['images'][0]['url']
+    relight_img_path = result[1]
 
     response = requests.get(relight_img_path, stream=True)