Update app.py

app.py

@@ -204,17 +204,21 @@ def load_model(checkpoint):
     if checkpoint == "tiny":
         sam2_checkpoint = "./checkpoints/sam2_hiera_tiny.pt"
         model_cfg = "sam2_hiera_t.yaml"
+        return sam2_checkpoint, model_cfg
     elif checkpoint == "samll":
         sam2_checkpoint = "./checkpoints/sam2_hiera_small.pt"
         model_cfg = "sam2_hiera_s.yaml"
+        return sam2_checkpoint, model_cfg
     elif checkpoint == "base-plus":
         sam2_checkpoint = "./checkpoints/sam2_hiera_base_plus.pt"
         model_cfg = "sam2_hiera_b+.yaml"
+        return sam2_checkpoint, model_cfg
     elif checkpoint == "large":
         sam2_checkpoint = "./checkpoints/sam2_hiera_large.pt"
         model_cfg = "sam2_hiera_l.yaml"
+        return sam2_checkpoint, model_cfg
 
-    return sam2_checkpoint, model_cfg
+    
     
 def sam_process(input_first_frame_image, checkpoint, tracking_points, trackings_input_label, video_frames_dir):
     # 1. We need to preprocess the video and store frames in the right directory