modify to use model_id from key

sidhulyalkar · Dec 14, 2023 · 8aca21c · 8aca21c
1 parent bf227e2
commit 8aca21c
Showing 1 changed file with 3 additions and 7 deletions.
diff --git a/element_facemap/facemap_inference.py b/element_facemap/facemap_inference.py
@@ -384,13 +384,9 @@ def make(self, key):
 
             bbox = (FacemapPoseEstimationTask & key).fetch1("bbox") or []
 
-            # Model Name of interest should be specified by user during facemap task params manual update
-            model_id = (FacemapPoseEstimationTask & key).fetch("model_id")
-
-            # Fetch model(.pt) file attachment to present working directory
-            facemap_model_name = (FacemapModel.File & f'model_id="{model_id}"').fetch1(
-                "model_file"
-            )
+            facemap_model_name = (
+                FacemapModel.File & f'model_id="{key["model_id"]}"'
+            ).fetch1("model_file")
 
             facemap_model_path = Path.cwd() / facemap_model_name
             models_root_dir = model_loader.get_models_dir()