Update gradio_demo.py

gradio_demo.py

@@ -972,6 +972,7 @@ def process_image(input_image, input_text):
         # input_boxes = np.array(input_boxes)
         # class_ids = np.array(class_ids)
 
+        predictions = task.result.objects
         classes = [x.strip().lower() for x in input_text.split('.') if x]
         class_name_to_id = {name: id for id, name in enumerate(classes)}
         class_id_to_name = {id: name for name, id in class_name_to_id.items()}
@@ -1487,7 +1488,7 @@ with block:
         
         return result
     
-    ips_bg = [input_fg, input_bg, prompt, image_width, image_height, num_samples, seed, steps, a_prompt, n_prompt, cfg, highres_scale, highres_denoise, bg_source, x_slider, y_slider, fg_scale_slider]
+    ips_bg = [input_fg, bg_manager.original_bg, prompt, image_width, image_height, num_samples, seed, steps, a_prompt, n_prompt, cfg, highres_scale, highres_denoise, bg_source, x_slider, y_slider, fg_scale_slider]
 
     # Update button click events with new inputs list
     relight_button_bg.click(