diff --git a/AIMeiSheng/docker_demo/common.py b/AIMeiSheng/docker_demo/common.py
index 7602d60..9d0ebff 100644
--- a/AIMeiSheng/docker_demo/common.py
+++ b/AIMeiSheng/docker_demo/common.py
@@ -1,108 +1,112 @@
 import os
+import sys
 import time
 # import logging
 import urllib, urllib.request
 
 # 测试/正式环境
-gs_prod = True
+gs_prod = False
+if len(sys.argv) > 1 and sys.argv[1] == "prod":
+    gs_prod = True
+print(gs_prod)
 
 gs_tmp_dir = "/tmp/ai_meisheng_tmp"
 gs_model_dir = "/tmp/ai_meisheng_models"
 gs_resource_cache_dir = "/tmp/ai_meisheng_resource_cache"
 gs_embed_model_path = os.path.join(gs_model_dir, "RawNet3/models/weights/model.pt")
 gs_svc_model_path = os.path.join(gs_model_dir,
                                  "weights/xusong_v2_org_version_alldata_embed_spkenx200x_vocal_e22_s95040.pth")
 gs_hubert_model_path = os.path.join(gs_model_dir, "hubert.pt")
 gs_rmvpe_model_path = os.path.join(gs_model_dir, "rmvpe.pt")
 gs_embed_model_spk_path = os.path.join(gs_model_dir, "SpeakerEncoder/pretrained_model/best_model.pth.tar")
-gs_embed_config_spk_path = os.path.join(gs_model_dir, "SpeakerEncoder/pretrained_model/config.json") 
+gs_embed_config_spk_path = os.path.join(gs_model_dir, "SpeakerEncoder/pretrained_model/config.json")
 
 # errcode
 
 gs_err_code_success = 0
 gs_err_code_download_vocal = 100
 gs_err_code_download_svc_url = 101
 gs_err_code_svc_process = 102
 gs_err_code_transcode = 103
 gs_err_code_volume_adjust = 104
 gs_err_code_upload = 105
 gs_err_code_params = 106
 gs_err_code_pending = 107
 gs_err_code_target_silence = 108
 gs_err_code_too_many_connections = 429
 
 gs_redis_conf = {
     "host": "av-credis.starmaker.co",
     "port": 6379,
     "pwd": "lKoWEhz%jxTO",
 }
 
 gs_server_redis_conf = {
     "producer": "test_ai_meisheng_producer",  # 输入的队列
     "ai_meisheng_key_prefix": "test_ai_meisheng_key_",  # 存储结果情况
 }
 
 if gs_prod:
     gs_server_redis_conf = {
         "producer": "ai_meisheng_producer",  # 输入的队列
         "ai_meisheng_key_prefix": "ai_meisheng_key_",  # 存储结果情况
     }
 
 
 def download2disk(url, dst_path):
     try:
         urllib.request.urlretrieve(url, dst_path)
         return os.path.exists(dst_path)
     except Exception as ex:
         print(f"download url={url} error", ex)
         return False
 
 
 def exec_cmd(cmd):
     # gs_logger.info(cmd)
     print(cmd)
     ret = os.system(cmd)
     if ret != 0:
         return False
     return True
 
 
 def exec_cmd_and_result(cmd):
     r = os.popen(cmd)
     text = r.read()
     r.close()
     return text
 
 
 def upload_file2cos(key, file_path, region='ap-singapore', bucket_name='av-audit-sync-sg-1256122840'):
     """
     将文件上传到cos
     :param key: 桶上的具体地址
     :param file_path: 本地文件地址
     :param region: 区域
     :param bucket_name: 桶地址
     :return:
     """
     gs_coscmd = "coscmd"
     gs_coscmd_conf = "~/.cos.conf"
 
     cmd = "{} -c {} -r {} -b {} upload {} {}".format(gs_coscmd, gs_coscmd_conf, region, bucket_name, file_path, key)
     if exec_cmd(cmd):
         cmd = "{} -c {} -r {} -b {} info {}".format(gs_coscmd, gs_coscmd_conf, region, bucket_name, key) \
               + "| grep  Content-Length |awk \'{print $2}\'"
         res_str = exec_cmd_and_result(cmd)
         # logging.info("{},res={}".format(key, res_str))
         size = float(res_str)
         if size > 0:
             return True
         return False
     return False
 
 
 def check_input(input_data):
     key_list = ["record_song_url", "target_url", "start", "end", "vocal_loudness", "female_recording_url",
                 "male_recording_url"]
     for key in key_list:
         if key not in input_data.keys():
             return False
     return True