Upload gradio_app.py

gradio_app.py

@@ -152,6 +152,8 @@ def prepare_ds_renderer(
     gr.Info("Preparing for Rendering...")
     mesh_renderer = MeshRendererPyTorch3D(mesh, DEVICE, config=None)
     extension = f"lesion_{num_lesion}"
+    # if texture_name != "No Lesion":
+    #     extension = f"{texture_name.lower().split(' ')[0]}_lesion_{num_lesion}"
     nevi_exists = os.path.exists(bodytex.annotation_filepath(mesh_name.split("_")[0]))
     gen2d = Generate2DHelper(
         mesh_filename=mesh_filename,
@@ -175,7 +177,7 @@ def prepare_ds_renderer(
         device=DEVICE,
         dir_blended_textures=dir_blended_textures,
         dir_anatomy=dir_anatomy,
-        extension=extension if num_lesion > 0 else "demo",
+        extension=extension ,
     )
     normal_texture = load_texture_map(
         mesh, mesh_name, "No Lesion", 0, device
@@ -249,7 +251,7 @@ def prepare_ds_renderer(
             gr.Info("Rasterization in progress...")
             gen2d.mesh_renderer.compute_fragments()
             gr.Info("Successfully rasterized.")
-        paste_img, target = gen2d.render_image_and_target(paste_lesion=True)
+        paste_img, target = gen2d.render_image_and_target(paste_lesion=False)
         if paste_img is None:
             gr.Info(
                 "***Not enough skin or unable to paste lesion. Skipping and Retrying."