Update flask_api_full_song.py

flask_api_full_song.py

@@ -6,6 +6,9 @@ from flask import Flask, request, send_file
 
 from inference import infer_tool, slicer
 
+import requests
+import os
+
 app = Flask(__name__)
 
 
@@ -49,7 +52,34 @@ def wav2wav():
 
 
 if __name__ == '__main__':
-    model_name = "logs/44k/G_60000.pth"  # 模型地址
-    config_name = "configs/config.json"  # config地址
+    secret_id = os.getenv('SECRET_ID')
+    secret_key = os.getenv('SECRET_KEY')
+    region = 'na-siliconvalley'
+    bucket_name = 'xiaohei-cat-ai-1304646510'
+    cos_path = '你要下载的文件在COS中的路径'
+
+    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)