cleanup readme and raise errors

README copy.md

@@ -1,12 +0,0 @@
----
-title: Find3D
-emoji: 🔥
-colorFrom: gray
-colorTo: indigo
-sdk: gradio
-sdk_version: 5.5.0
-app_file: app.py
-pinned: false
----
-
-Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

app.py

@@ -46,10 +46,14 @@ def predict(pcd_path, inference_mode, part_queries):
     xyz, rgb, normal = read_pcd(pcd_path)
     if inference_mode == "Segmentation":
         parts = [part.strip(" ") for part in re.split(r'[,;.|]', part_queries)]
+        if len(parts)< 2:
+            raise gr.Error("For segmentation mode, please provide 2 or more parts", duration=5)
         seg_rgb = segment_obj(xyz, rgb, normal, parts).cpu().numpy()
         legend = get_legend(parts)
         return render_point_cloud(xyz, seg_rgb, legend=legend)
     elif inference_mode == "Localization":
+        if "," in part_queries or ";" in part_queries or "." in part_queries:
+            raise gr.Error("For localization mode, please provide only one part", duration=5)
         heatmap_rgb = get_heatmap(xyz, rgb, normal, part_queries).cpu().numpy()
         return render_point_cloud(xyz, heatmap_rgb)
     else:
@@ -64,10 +68,11 @@ def on_select(evt: gr.SelectData):
 with gr.Blocks(theme=gr.themes.Default(text_size="lg", radius_size="none")) as demo:
     gr.HTML(
         '''<h1 text-align="center">Find Any Part in 3D</h1>
-        <p style='font-size: 16px;'>This is a demo for Find3D: Find Any Part in 3D! Two modes are supported: segmentation and localization.
-        For segmentation mode, please provide multiple part queries in the "queries" text box, in the format of comma-separated string, such as "part1,part2,part3".
+        <p style='font-size: 16px;'>This is a demo for Find3D: Find Any Part in 3D! Two modes are supported: <b>segmentation</b> and <b>localization</b>.
+        For <b>segmentation mode</b>, please provide multiple part queries in the "queries" text box, in the format of comma-separated string, such as "part1,part2,part3".
         After hitting "Run", the model will segment the object into the provided parts.
-        For localization mode, please only provide one query string in the "queries" text box. After hitting "Run", the model will generate a heatmap for the provided query text.
+        <br>
+        For <b>localization mode<b>, please only provide one query string in the "queries" text box. After hitting "Run", the model will generate a heatmap for the provided query text.
         Please click on the buttons below "Objaverse" and "In the Wild" for some examples. You can also upload your own .pcd files.</p>
         <p style='font-size: 16px;'>Hint: 
         When uploading your own point cloud, please first close the existing point cloud by clicking on the "x" button.