Move config file to model repository

python_utils/get_model.py

@@ -1,4 +1,6 @@
 import urllib.request
+import tempfile
+
 
 OPTIMAL_NMS_THRESHOLD = 0.7
 _model_config_url = "https://huggingface.co/TZProject/final_tz_segmentor/resolve/main/final_model_config.yaml"
@@ -21,9 +23,10 @@ def load_model():
     ## define relevant parameters
     cfg = get_cfg()
 
-    config_filename = "./final_model_config.yaml"
-    urllib.request.urlretrieve(_model_config_url, filename=config_filename)
-    cfg.merge_from_file(config_filename)
+    with tempfile.NamedTemporaryFile(suffix=".yaml") as tmp:
+        print(tmp.name)
+        urllib.request.urlretrieve(_model_config_url, filename=tmp.name)
+        cfg.merge_from_file(tmp.name)
     if not torch.cuda.is_available():
         cfg.MODEL.DEVICE = "cpu"
     else: