carekit

app.py

@@ -131,10 +131,26 @@ def sam_init():
     predictor = SamPredictor(sam)
     return predictor
 
-@spaces.GPU
-def create_carvekit_interface():
-    # Check doc strings for more information
-    interface = HiInterface(object_type="object",  # Can be "object" or "hairs-like".
+# @spaces.GPU
+# def create_carvekit_interface():
+#     # Check doc strings for more information
+#     interface = HiInterface(object_type="object",  # Can be "object" or "hairs-like".
+#                             batch_size_seg=6,
+#                             batch_size_matting=1,
+#                             device=device,
+#                             seg_mask_size=640,  # Use 640 for Tracer B7 and 320 for U2Net
+#                             matting_mask_size=2048,
+#                             trimap_prob_threshold=231,
+#                             trimap_dilation=30,
+#                             trimap_erosion_iters=5,
+#                             fp16=True)
+#
+#     return interface
+
+
+# rembg_session = rembg.new_session()
+# rembg_session = create_carvekit_interface()
+interface = HiInterface(object_type="object",  # Can be "object" or "hairs-like".
                             batch_size_seg=6,
                             batch_size_matting=1,
                             device=device,
@@ -144,12 +160,7 @@ def create_carvekit_interface():
                             trimap_dilation=30,
                             trimap_erosion_iters=5,
                             fp16=True)
-
-    return interface
-
-
-# rembg_session = rembg.new_session()
-rembg_session = create_carvekit_interface()
+rembg_session = interface
 predictor = sam_init()