update

app.py

@@ -41,12 +41,23 @@ with gr.Blocks(
                         type="pil",
                         visible=False,
                     )
+
+                    css = """
+                    <style>
+                    #img1 img {
+                        object-fit: contain !important;
+                        width: 100% !important;
+                        height: 100% !important;
+                    }
+                    </style>
+                    """
                     image_prompt = gr.Image(
                         label="Input Image",
                         format="png",
                         image_mode="RGBA",
                         type="pil",
-                        height=400,
+                        height=300,
+                        elem_id="img1",
                     )
                     gr.Markdown(
                         """
@@ -59,7 +70,7 @@ with gr.Blocks(
                     with gr.Row():
                         with gr.Column(scale=1):
                             image_prompt_sam = gr.Image(
-                                label="Input Image", type="numpy", height=400
+                                label="Input Image", type="numpy", height=500
                             )
                             image_seg_sam = gr.Image(
                                 label="SAM Seg Image",
@@ -445,4 +456,4 @@ with gr.Blocks(
 
 if __name__ == "__main__":
     demo.launch()
-    
+

common.py

@@ -208,7 +208,7 @@ def preprocess_image_fn(
     return image, image_cache
 
 
-@spaces.GPU
+# @spaces.GPU
 def preprocess_sam_image_fn(
     image: Image.Image
 ) -> tuple[Image.Image, Image.Image]:
@@ -251,6 +251,8 @@ def get_selected_image(
 
 
 def get_cached_image(image_path: str) -> Image.Image:
+    if isinstance(image_path, Image.Image):
+        return image_path
     return Image.open(image_path).resize((512, 512))