Update app.py

app.py

@@ -124,11 +124,11 @@ def create_legend_image(labels_to_ids: dict[str, int], filename="legend.png"):
 
 # ----------------- MODEL ----------------- #
 
-URL = "https://huggingface.co/facebook/sapiens/resolve/main/sapiens_lite_host/torchscript/normal/checkpoints/sapiens_1b/sapiens_1b_normal_render_people_epoch_115_torchscript.pt2?download=true"
+URL = "https://huggingface.co/facebook/sapiens/resolve/main/sapiens_lite_host/torchscript/depth/checkpoints/sapiens_1b/sapiens_1b_render_people_epoch_88_torchscript.pt2?download=true"
 CHECKPOINTS_DIR = os.path.join(ASSETS_DIR, "checkpoints")
 os.makedirs(CHECKPOINTS_DIR, exist_ok=True)
 
-model_path = os.path.join(CHECKPOINTS_DIR, "sapiens_1b_normal_render_people_epoch_115_torchscript.pt2")
+model_path = os.path.join(CHECKPOINTS_DIR, "sapiens_1b_render_people_epoch_88_torchscript.pt2")
 
 if not os.path.exists(model_path) or os.path.getsize(model_path) == 0:
     print("Downloading model...")
@@ -192,7 +192,7 @@ with gr.Blocks(css=CUSTOM_CSS, theme=gr.themes.Monochrome(radius_size=sizes.radi
         with gr.Column():
             input_image = gr.Image(label="Input Image", type="pil", format="png")
         with gr.Column():
-            result_image = gr.Image(label="Normal Output", format="png")
+            result_image = gr.Image(label="Depth Output", format="png")
             run_button = gr.Button("Run")
 
     run_button.click(