Update flask_api_full_song.py

flask_api_full_song.py

@@ -11,6 +11,38 @@ import threading
 from qcloud_cos import CosConfig
 from qcloud_cos import CosS3Client
 
+
+secret_id = os.getenv('SECRET_ID')
+secret_key = os.getenv('SECRET_KEY')
+region = 'na-siliconvalley'
+bucket_name = 'xiaohei-cat-ai-1304646510'
+
+config = CosConfig(Region=region, SecretId=secret_id, SecretKey=secret_key)
+client = CosS3Client(config)
+
+response1 = client.get_object(
+    Bucket=bucket_name,
+    Key="models/So-VITS-SVC/Koxia-Full/G_full.pth"
+)
+
+with open("/tmp/G_full.pth", 'wb') as local_file:
+    for chunk in response1['Body'].iter_chunks():
+        local_file.write(chunk)
+
+response2 = client.get_object(
+    Bucket=bucket_name,
+    Key="models/So-VITS-SVC/Koxia-Full/config.json"
+)
+
+with open("/tmp/config.json", 'wb') as local_file:
+    for chunk in response2['Body'].iter_chunks():
+        local_file.write(chunk)
+
+model_name = "/tmp/G_full.pth"  # 模型地址
+config_name = "/tmp/config.json"  # config地址
+svc_model = infer_tool.Svc(model_name, config_name)
+
+
 tasks = {}
 running_threads = 0
 condition = threading.Condition()
@@ -105,33 +137,4 @@ def download(filename):
 
 
 if __name__ == '__main__':
-    secret_id = os.getenv('SECRET_ID')
-    secret_key = os.getenv('SECRET_KEY')
-    region = 'na-siliconvalley'
-    bucket_name = 'xiaohei-cat-ai-1304646510'
-
-    config = CosConfig(Region=region, SecretId=secret_id, SecretKey=secret_key)
-    client = CosS3Client(config)
-
-    response1 = client.get_object(
-        Bucket=bucket_name,
-        Key="models/So-VITS-SVC/Koxia-Full/G_full.pth"
-    )
-
-    with open("/tmp/G_full.pth", 'wb') as local_file:
-        for chunk in response1['Body'].iter_chunks():
-            local_file.write(chunk)
-
-    response2 = client.get_object(
-        Bucket=bucket_name,
-        Key="models/So-VITS-SVC/Koxia-Full/config.json"
-    )
-
-    with open("/tmp/config.json", 'wb') as local_file:
-        for chunk in response2['Body'].iter_chunks():
-            local_file.write(chunk)
-    
-    model_name = "/tmp/G_full.pth"  # 模型地址
-    config_name = "/tmp/config.json"  # config地址
-    svc_model = infer_tool.Svc(model_name, config_name)
     app.run(port=1145, host="0.0.0.0", debug=False, threaded=False)