update

app.py

@@ -46,8 +46,6 @@ with gr.Blocks(
                     <style>
                     #img-fit img {
                         object-fit: contain !important;
-                        width: 100% !important;
-                        height: 100% !important;
                     }
                     </style>
                     """

common.py

@@ -80,7 +80,7 @@ IMAGESR_MODEL = ImageRealESRGAN(outscale=4)
 
 if os.getenv("GRADIO_APP") == "imageto3d":
     RBG_REMOVER = RembgRemover()
-    SAM_PREDICTOR = SAMPredictor(model_type="vit_h")
+    SAM_PREDICTOR = SAMPredictor(model_type="vit_h", device="cpu")
     PIPELINE = TrellisImageTo3DPipeline.from_pretrained(
         "JeffreyXiang/TRELLIS-image-large"
     )
@@ -319,6 +319,8 @@ def select_point(
         sel_pix.append((evt.index, 1))  # default foreground_point
 
     masks = SAM_PREDICTOR.generate_masks(image, sel_pix)
+    print("sel_pix", sel_pix)
+    print("masks", masks)
     seg_image = SAM_PREDICTOR.get_segmented_image(image, masks)
 
     for point, label in sel_pix: