Update app.py

app.py

@@ -40,18 +40,18 @@ def main():
     save_dir = Path(SAVE_DIR)
     save_dir.mkdir(parents=True, exist_ok=True)
 
-    def pose(img_path, bone_lengths=True, foreshortening=True, self_contacts=False, naturalness=True):
-        if bone_lengths == False:
-            foreshortening = False
+    def pose(img_path, use_cos=True, use_angle_transf=True, use_contacts=False, use_natural=True):
+        if use_cos == False:
+            use_angle_transf = False
 
         cmd = CMD.format(save_dir, img_path)
-        if bone_lengths:
+        if use_cos:
             cmd = cmd + " --use-cos"
-        if foreshortening:
+        if use_angle_transf:
             cmd = cmd + " --use-angle-transf"
-        if self_contacts:
+        if use_contacts:
             cmd = cmd + " --use-contacts"
-        if naturalness:
+        if use_natural:
             cmd = cmd + " --use-natural"
 
         out_dir = (save_dir / Path(img_path).name).with_suffix("")
@@ -64,19 +64,19 @@ def main():
 
     examples = []
     for img_path in Path("./data/images").glob("*"):
-        examples.append([str(img_path), True, True, True, True])
+        examples.append([str(img_path), True, True, False, True])
         break
 
     demo = gr.Interface(
         fn=pose,
         inputs=[
-            gr.Image(type="filepath"),
-            gr.Checkbox(value=True),
-            gr.Checkbox(value=True),
-            gr.Checkbox(value=False, interactive=torch.cuda.is_available()),
-            gr.Checkbox(value=True),
+            gr.Image(type="filepath", label="Image"),
+            gr.Checkbox(value=True, label="Bone lenghts"),
+            gr.Checkbox(value=True, label="Foreshortening"),
+            gr.Checkbox(value=False, label="Self-contacts (available with cuda)", interactive=torch.cuda.is_available()),
+            gr.Checkbox(value=True, label="Pose naturalness"),
         ],
-        outputs=gr.Model3D(),
+        outputs=gr.Model3D(clear_color=[0.0, 0.0, 0.0, 0.0], label="SMPL 3D pose"),
         examples=examples,
     )