Update flask_api_full_song.py

flask_api_full_song.py

@@ -8,6 +8,8 @@ from inference import infer_tool, slicer
 
 import requests
 import os
+from qcloud_cos import CosConfig
+from qcloud_cos import CosS3Client
 
 app = Flask(__name__)
 
@@ -56,7 +58,6 @@ if __name__ == '__main__':
     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)
@@ -67,8 +68,8 @@ if __name__ == '__main__':
     )
 
     with open("/tmp/G_full.pth", 'wb') as local_file:
-    for chunk in response1['Body'].iter_chunks():
-        local_file.write(chunk)
+        for chunk in response1['Body'].iter_chunks():
+            local_file.write(chunk)
 
     response2 = client.get_object(
         Bucket=bucket_name,
@@ -76,10 +77,10 @@ if __name__ == '__main__':
     )
 
     with open("/tmp/config.json", 'wb') as local_file:
-    for chunk in response2['Body'].iter_chunks():
-        local_file.write(chunk)
+        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)
+    app.run(port=1145, host="0.0.0.0", debug=False, threaded=False)