fix:asymmetry

app.py

@@ -62,7 +62,7 @@ def generate_func(
 ):
     # geometry_model = geometry_model.to("cuda")
     if "Label" in args.geometry_model:
-        symmetry_values = ['x', 'asymmetry']
+        symmetry_values = ["x", "asymmetry"]
         out = geometry_model(
             input_image_path,
             label={"symmetry": symmetry_values[int(symmetry)], "edge_type": edge_type},

step1x3d_geometry/models/conditional_encoders/label_encoder.py

@@ -120,7 +120,7 @@ class LabelEncoder(BaseLabelEncoder, ModelMixin):
                     symmetry_type_label_embeds.append(
                         self.embedding_table_symmetry_type(
                             torch.tensor(
-                                SYMMETRY_TYPE_MAPPING[label["symmetry"][0]]
+                                SYMMETRY_TYPE_MAPPING[label["symmetry"]]
                             ).to(get_device())
                         )
                     )