sym

app.py

@@ -48,9 +48,10 @@ def generate_func(
 ):
     geometry_model = geometry_model.to("cuda")
     if "Label" in args.geometry_model:
+        symmetry_values = ['x', 'asymmetry']
         out = geometry_model(
             input_image_path,
-            label={"symmetry": symmetry, "edge_type": edge_type},
+            label={"symmetry": symmetry_values[int(symmetry)], "edge_type": edge_type},
             guidance_scale=float(guidance_scale),
             octree_resolution=384,
             max_facenum=int(max_facenum),
@@ -113,10 +114,10 @@ if __name__ == "__main__":
                 )
                 max_facenum = gr.Number(label="Max Face Num", value="400000")
                 symmetry = gr.Radio(
-                    choices=["x", "asymmetry"],
+                    choices=["symmetry", "asymmetry"],
                     label="Symmetry Type",
-                    value="x",
-                    type="value",
+                    value="symmetry",
+                    type="index",
                 )
                 edge_type = gr.Radio(
                     choices=["sharp", "normal", "smooth"],