Merge pull request #35 from kevinlin09/feat/cv_model_call

feat(cv): support model calling

Author: kevinlin09

dashscope/utils/oss_utils.py

@@ -157,6 +157,7 @@ def check_and_upload_local(model: str, content: str, api_key: str):
     elif content.startswith('oss://'):
         return True, content
     elif not content.startswith('http'):
+        content = os.path.expanduser(content)
         if os.path.isfile(content):
             file_url = OssUtils.upload(model=model,
                                        file_path=content,

samples/test_cv_models.py

@@ -0,0 +1,32 @@
+import os
+
+from dashscope.client.base_api import BaseAsyncApi
+from dashscope.utils.oss_utils import check_and_upload_local
+
+# style repaint (ref: https://help.aliyun.com/zh/model-studio/portrait-style-redraw-api-reference)
+
+api_key = os.getenv('DASHSCOPE_API_KEY')
+model = 'wanx-style-repaint-v1'
+file_path = '~/Downloads/cat.png'
+
+uploaded, image_url = check_and_upload_local(model=model, content=file_path, api_key=api_key)
+
+kwargs = {}
+if uploaded is True:
+    headers = {'X-DashScope-OssResourceResolve': 'enable'}
+    kwargs['headers'] = headers
+
+response = BaseAsyncApi.call(
+    model=model,
+    input={
+        # "image_url": "https://public-vigen-video.oss-cn-shanghai.aliyuncs.com/public/dashscope/test.png",
+        "image_url": image_url,
+        "style_index":3
+    },
+    task_group='aigc',
+    task='image-generation',
+    function='generation',
+    **kwargs
+)
+
+print('response: \n%s\n' % response)

samples/test_image_synthesis.py

@@ -13,8 +13,7 @@
                           model="wanx2.1-t2i-turbo",
                           prompt=prompt,
                           n=1,
-                          size='1024*1024',
-                          enable_encryption=True)
+                          size='1024*1024')
 
 print('response: %s' % rsp)
 if rsp.status_code == HTTPStatus.OK: