Update app.py

app.py

@@ -12,13 +12,13 @@ import tempfile
 from classes_and_palettes import GOLIATH_PALETTE, GOLIATH_CLASSES
 
 class Config:
-    os.system(f"wget https://huggingface.co/shimu0215/seg/resolve/main/sapiens_0.3b_goliath_best_goliath_mIoU_7673_epoch_194_torchscript.pt2")
+    # 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")
+    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",
@@ -28,7 +28,7 @@ class Config:
 class ModelManager:
     @staticmethod
     def load_model(checkpoint_name: str):
-        checkpoint_path = os.path.join(Config.CHECKPOINTS_DIR, Config.CHECKPOINTS[checkpoint_name])
+        checkpoint_path = os.path.join(‘shimu0215/seg/’, Config.CHECKPOINTS[checkpoint_name])
         model = torch.jit.load(checkpoint_path)
         model.eval()
         model.to("cuda")