start of space

README.md

@@ -1,8 +1,8 @@
 ---
-title: PrivateCutLER
+title: CutS3D
 emoji: 🌖
-colorFrom: yellow
-colorTo: green
+colorFrom: red
+colorTo: blue
 sdk: docker
 pinned: false
 license: mit

app.py

@@ -6,7 +6,7 @@ import gradio as gr
 
 from model import run_model
 
-DESCRIPTION = '# [CutLER](https://github.com/facebookresearch/CutLER)'
+DESCRIPTION = '# [CutS3D](https://leonsick.github.io/cuts3d/) Cutting Semantics in 3D for 2D Unsupervised Instance Segmentation'
 
 paths = sorted(pathlib.Path('CutLER/cutler/demo/imgs').glob('*.jpg'))
 

model.py

@@ -97,8 +97,6 @@ def load_model(score_threshold: float) -> VisualizationDemo:
     #weight_path = model_dir / WEIGHT_URL.split('/')[-1]
     weight_path = model_dir / ".cache" / "huggingface" / "download"/ model_filename
     weight_path = model_dir / model_filename
-    print("***weight_path***", weight_path)
-    print("torch.cuda.is_available()", torch.cuda.is_available())
 
 
     # Load the model weights file from huggingface hub, use token to authenticate