rootglitch commited on
Commit
2f0dff5
·
1 Parent(s): 39bd3ee

Return the 2nd AI image

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -514,7 +514,7 @@ def generate_ai_bg(input_img, prompt):
514
  )
515
  print(result)
516
 
517
- relight_img_path = result[0][0]['image']
518
 
519
  # response = requests.get(relight_img_path, stream=True)
520
  relight_img = Image.open(relight_img_path).convert("RGBA")
 
514
  )
515
  print(result)
516
 
517
+ relight_img_path = result[0][1]['image']
518
 
519
  # response = requests.get(relight_img_path, stream=True)
520
  relight_img = Image.open(relight_img_path).convert("RGBA")