Update app.py

app.py

@@ -12,8 +12,13 @@ import tempfile
 from classes_and_palettes import GOLIATH_PALETTE, GOLIATH_CLASSES
 
 class Config:
-    ASSETS_DIR = os.path.join(os.path.dirname(__file__), 'assets')
-    CHECKPOINTS_DIR = os.path.join(ASSETS_DIR, "checkpoints")
+    os.system(f"wget https://huggingface.co/shimu0215/seg/resolve/main/sapiens_0.3b_goliath_best_goliath_mIoU_7673_epoch_194_torchscript.pt2")
+
+    
+    # ASSETS_DIR = os.path.join(os.path.dirname(__file__), 'assets')
+    # CHECKPOINTS_DIR = os.path.join(ASSETS_DIR, "checkpoints")
+    ASSETS_DIR = os.system(f"wget https://huggingface.co/shimu0215/seg/resolve/")
+    CHECKPOINTS_DIR = os.path.join(ASSETS_DIR, "main")
     CHECKPOINTS = {
         "0.3b": "sapiens_0.3b_goliath_best_goliath_mIoU_7673_epoch_194_torchscript.pt2",
         "0.6b": "sapiens_0.6b_goliath_best_goliath_mIoU_7777_epoch_178_torchscript.pt2",