Externalize model_name

bo-lu · bo-lu · commit dceb3d89379c · 2024-12-31T11:29:29.000-05:00
diff --git a/deployment/lambda-search/app.py b/deployment/lambda-search/app.py
@@ -15,7 +15,7 @@
 aos_host = environ['OS_ENDPOINT'] 
 sagemaker_endpoint = environ['SAGEMAKER_ENDPOINT'] 
 os_secret_id = environ['OS_SECRET_ID']
-model_name = 'minilm-knn'
+model_name = environ['MODEL_NAME']
 
 def get_awsauth_from_secret(region, secret_id):
     """
