Update app.py

app.py

@@ -9,6 +9,7 @@ from torchvision import transforms
 from PIL import Image
 import matplotlib.pyplot as plt
 import tempfile
+from huggingface_hub import hf_hub_download
 
 class Config:
     ASSETS_DIR = os.path.join(os.path.dirname(__file__), 'assets')
@@ -28,10 +29,11 @@ class Config:
 class ModelManager:
     @staticmethod
     def load_model(checkpoint_name: str):
-        if checkpoint_name is None:
-            return None
-        checkpoint_path = os.path.join(Config.CHECKPOINTS_DIR, checkpoint_name)
-        model = torch.jit.load(checkpoint_path)
+       model_path = hf_hub_download(
+            repo_id="shimu0215/seg", 
+            filename="sapiens_2b_render_people_epoch_25_torchscript.pt2",    
+        )
+        model = torch.jit.load(model_path)
         model.eval()
         model.to("cuda")
         return model