liqian hace 1 año
padre
commit
1ae7d12663
Se han modificado 2 ficheros con 4 adiciones y 4 borrados
  1. 2 2
      rec_model_config.py
  2. 2 2
      rec_model_update.py

+ 2 - 2
rec_model_config.py

@@ -3,10 +3,10 @@ OSS_CONFIG = {
     'endpoint': 'http://oss-cn-hangzhou-internal.aliyuncs.com',
     'accessKeyId': 'LTAIP6x1l3DXfSxm',
     'accessKeySecret': 'KbTaM9ars4OX3PMS6Xm7rtxGr1FLon',
-    'bucket_name': ''
+    'bucket_name': 'recommend-sort-models'
 }
 
-slb_id = ''
+slb_id = 'lb-bp1ho0fp6nh7j93qij1kx'
 
 # 修改负载均衡权限
 slb_client_params = {

+ 2 - 2
rec_model_update.py

@@ -43,7 +43,7 @@ def rec_server_health_check(client, instance_id):
     global health_instances
     ip_address = utils.get_ip_address(client=client, instance_id=instance_id)
     while True:
-        health_check_url = f"http://{ip_address}:5001/healthcheck"
+        health_check_url = f"http://{ip_address}:8501/v1/models/deepfm"
         try:
             http_code = requests.get(health_check_url).status_code
         except:
@@ -129,7 +129,7 @@ def main():
             logging.info(f"online instance ids: {online_instance_ids}")
 
             # 3. 逐台更新
-            image_name = ''
+            image_name = 'tensorflow/serving'
             for i, instance_id in enumerate(online_instance_ids):
                 logging.info(f"instance:{instance_id}")
                 # 3.1. 获取ip