Update merged_app2.py

merged_app2.py

@@ -863,6 +863,8 @@ def process(input_fg, prompt, image_width, image_height, num_samples, seed, step
 
     return pixels
 
+@spaces.GPU(duration=60)
+@torch.inference_mode()
 def extract_foreground(image):
     if image is None:
         return None, gr.update(visible=True), gr.update(visible=True)
@@ -1864,11 +1866,29 @@ with gr.Blocks() as app:
                 output_image.render()
                 transfer_btn = gr.Button("Send to relight")
 
+    with gr.Tab("Caption"):
+        with gr.Row():
+            gr.Markdown("## Describe Image")
+        with gr.Row():
+            with gr.Column():
+                with gr.Row():
+                    image_to_describe = gr.Image(type="pil", label="Image", height=480)
+                with gr.Row():
+                    with gr.Group():
+                        describe_button = gr.Button(value="Describe Image")
+                        description_text = gr.Textbox(
+                                label="Output", 
+                                placeholder="",
+                                value=""
+                            )
+
         def send_img(img_result):
                 return img_result
         
         transfer_btn.click(send_img, [output_image], [input_fg])
 
+        # describe_button.click(describe_image, [image_to_describe], [description_text])
+
                 
         generate_btn.click(
             fn=generate_image,