Update app.py

app.py

@@ -13,7 +13,7 @@ from torchvision import transforms
 
 # ----------------- HELPER FUNCTIONS ----------------- #
 
-ASSETS_DIR = os.path.join(os.path.dirname(__file__), "assets")
+#ASSETS_DIR = os.path.join(os.path.dirname(__file__), "assets")
 
 LABELS_TO_IDS = {
     "Background": 0,
@@ -132,9 +132,9 @@ 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/seg/checkpoints/sapiens_0.3b/sapiens_0.3b_goliath_best_goliath_mIoU_7673_epoch_194_torchscript.pt2?download=true"
-CHECKPOINTS_DIR = os.path.join(ASSETS_DIR, "checkpoints")
-model_path = os.path.join(CHECKPOINTS_DIR, "sapiens_0.3b_goliath_best_goliath_mIoU_7673_epoch_194_torchscript.pt2")
+URL = "https://huggingface.co/facebook/sapiens/resolve/main/sapiens_lite_host/torchscript/depth/checkpoints/sapiens_0.3b/sapiens_0.3b_render_people_epoch_100_torchscript.pt2?download=true"
+#CHECKPOINTS_DIR = os.path.join(ASSETS_DIR, "checkpoints")
+model_path = os.path.join(os.path.dirname(__file__), "sapiens_0.3b_goliath_best_goliath_mIoU_7673_epoch_194_torchscript.pt2")
 
 if not os.path.exists(model_path):
     os.makedirs(CHECKPOINTS_DIR, exist_ok=True)
@@ -199,19 +199,19 @@ 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")
 
-            example_model = gr.Examples(
+            '''example_model = gr.Examples(
                 inputs=input_image,
                 examples_per_page=10,
                 examples=[
                     os.path.join(ASSETS_DIR, "examples", img)
                     for img in os.listdir(os.path.join(ASSETS_DIR, "examples"))
                 ],
-            )
+            )'''
         with gr.Column():
             result_image = gr.Image(label="Segmentation Result", format="png")
             run_button = gr.Button("Run")
 
-            gr.Image(os.path.join(ASSETS_DIR, "legend.png"), label="Legend", type="filepath")
+            #gr.Image(os.path.join(ASSETS_DIR, "legend.png"), label="Legend", type="filepath")
 
     run_button.click(
         fn=segment,