diff --git a/cases/ai1.py b/cases/ai1.py new file mode 100644 index 00000000..ff92825e --- /dev/null +++ b/cases/ai1.py @@ -0,0 +1,12 @@ +from modelscope.pipelines import pipeline +from modelscope.outputs import OutputKeys +from PIL import Image +from background_generation import modelscope_warpper + +model = "damo/cv_background_generation_sd" +pipe = pipeline('background_generation_task', model=model, device='gpu', auto_collate=False, model_revision='v1.1.0') +main_image = 'https://vision-poster.oss-cn-shanghai.aliyuncs.com/lllcho.lc/data/test_data/demo_example/%E5%8C%96%E5%A6%86%E5%93%81/1c33fc5e8b084269ffdb4e0557c2c3c4.png' +reference_image = 'https://vision-poster.oss-cn-shanghai.aliyuncs.com/lllcho.lc/data/test_data/5d873b5f64b82bcbb235748347602dce38c6ec1d.jpg' +out = pipe(main_image, reference_image, num_images_per_prompt=1) +imgs = out[OutputKeys.OUTPUT_IMGS] +imgs[0].save(f'result.jpg') diff --git a/conv.php b/conv.php index 113c78be..d92fe962 100644 --- a/conv.php +++ b/conv.php @@ -119,6 +119,8 @@ class Translator return 'PyCore::str("' . $attr->s . '")'; case 'Call': return $this->parseCall($attr); + case 'Subscript': + return $this->parseSubscript($attr); default: var_dump(__METHOD__, __LINE__); debug($attr);