Update app_3.py

app_3.py

@@ -465,8 +465,8 @@ def resize_without_crop(image, target_width, target_height):
     resized_image = pil_image.resize((target_width, target_height), Image.LANCZOS)
     return np.array(resized_image)
 
-@spaces.GPU(duration=60)
-@torch.inference_mode()
+# @spaces.GPU(duration=60)
+# @torch.inference_mode()
 # def run_rmbg(img, sigma=0.0):
 #     # Convert RGBA to RGB if needed
 #     if img.shape[-1] == 4:
@@ -1196,7 +1196,7 @@ with block:
         with gr.Row():
             with gr.Column():
                 with gr.Row():
-                    input_fg = gr.Image(type="numpy", label="Image", height=480)
+                    input_fg = gr.Image(type="pil", label="Image", height=480)
                 with gr.Row():
                     with gr.Group():
                         find_objects_button = gr.Button(value="(Option 1) Segment Object from text")
@@ -1208,7 +1208,7 @@ with block:
                     extract_button = gr.Button(value="Remove Background")
                 with gr.Row():
                     extracted_objects = gr.Image(type="numpy", label="Extracted Foreground", height=480)
-                    extracted_fg = gr.Image(type="numpy", label="Extracted Foreground", height=480)
+                    extracted_fg = gr.Image(type="pil", label="Extracted Foreground", height=480)
                     angles_fg = gr.Image(type="pil", label="Converted Foreground", height=480, visible=False)
 
                 with gr.Row():