jhj0517 commited on
Commit
afb73a1
·
1 Parent(s): a5bb941

Add model initialization on start

Browse files
Files changed (1) hide show
  1. modules/sam_inference.py +4 -0
modules/sam_inference.py CHANGED
@@ -48,6 +48,10 @@ class SamInference:
48
  self.video_inference_state = None
49
  self.video_info = None
50
 
 
 
 
 
51
  def load_model(self,
52
  model_type: Optional[str] = None,
53
  load_video_predictor: bool = False):
 
48
  self.video_inference_state = None
49
  self.video_info = None
50
 
51
+ # Initialization on huggingface space
52
+ for model_name in self.available_models:
53
+ self.load_model(model_type=model_name)
54
+
55
  def load_model(self,
56
  model_type: Optional[str] = None,
57
  load_video_predictor: bool = False):