Update app.py

app.py

@@ -131,8 +131,9 @@ def main():
         return str(mesh_path)
 
     examples = []
+    use_contacts = torch.cuda.is_available()
     for img_path in Path("./data/images").glob("*"):
-        examples.append([str(img_path), True, True, True, True])
+        examples.append([str(img_path), True, True, use_contacts, True])
 
     demo = gr.Interface(
         fn=pose,
@@ -140,7 +141,7 @@ def main():
             gr.Image(type="filepath", label="Image"),
             gr.Checkbox(value=True, label="Bone lenghts"),
             gr.Checkbox(value=True, label="Foreshortening"),
-            gr.Checkbox(value=True, label="Self-contacts", interactive=torch.cuda.is_available()),
+            gr.Checkbox(value=use_contacts, label="Self-contacts", interactive=use_contacts),
             gr.Checkbox(value=True, label="Pose naturalness"),
         ],
         outputs=gr.Model3D(clear_color=[0.0, 0.0, 0.0, 0.0], label="SMPL 3D pose"),