Spaces:
Running
Running
jhj0517
commited on
Commit
·
afb73a1
1
Parent(s):
a5bb941
Add model initialization on start
Browse files- 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):
|