diff --git a/readme.md b/readme.md index d37a658..24913db 100644 --- a/readme.md +++ b/readme.md @@ -186,7 +186,7 @@ pip install scepter | Controllable Image Synthesis | [🌟SCEdit(CVPR24)](docs/en/tasks/scedit.md) | [![Arxiv link](https://img.shields.io/static/v1?label=arXiv&message=SCEdit&color=red&logo=arxiv)](https://arxiv.org/abs/2312.11392) [![Page link](https://img.shields.io/badge/Page-SCEdit-Gree)](https://scedit.github.io/) | | Image Editing | [🌟LAR-Gen](docs/en/tasks/largen.md) | [![Arxiv link](https://img.shields.io/static/v1?label=arXiv&message=LARGen&color=red&logo=arxiv)](https://arxiv.org/abs/2403.19534) [![Page link](https://img.shields.io/badge/Page-LARGen-Gree)](https://ali-vilab.github.io/largen-page/) | | Image Editing | [🌟StyleBooth](docs/en/tasks/stylebooth.md) | [![Arxiv link](https://img.shields.io/static/v1?label=arXiv&message=StyleBooth&color=red&logo=arxiv)](https://arxiv.org/abs/2404.12154) [![Page link](https://img.shields.io/badge/Page-StyleBooth-Gree)](https://ali-vilab.github.io/stylebooth-page/) | -| Image Generation and Editing | [🌟ACE](https://ali-vilab.github.io/ace-page/) | [![Arxiv link](https://img.shields.io/static/v1?label=arXiv&message=ACE&color=red&logo=arxiv)](https://arxiv.org/abs/2410.00086) [![Page link](https://img.shields.io/badge/Page-ACE-Gree)](https://ali-vilab.github.io/ace-page/) [![Demo link](https://img.shields.io/badge/Demo-ACE-purple)]()
[![ModelScope link](https://img.shields.io/badge/ModelScope-Model-blue)](https://www.modelscope.cn/models/iic/ACE-0.6B-512px) [![HuggingFace link](https://img.shields.io/badge/%F0%9F%A4%97%20Hugging%20Face-Model-yellow)](https://huggingface.co/scepter-studio/ACE-0.6B-512px) | +| Image Generation and Editing | [🌟ACE](https://ali-vilab.github.io/ace-page/) | [![Arxiv link](https://img.shields.io/static/v1?label=arXiv&message=ACE&color=red&logo=arxiv)](https://arxiv.org/abs/2410.00086) [![Page link](https://img.shields.io/badge/Page-ACE-Gree)](https://ali-vilab.github.io/ace-page/) [![Demo link](https://img.shields.io/badge/Demo-ACE-purple)](https://huggingface.co/spaces/scepter-studio/ACE-Chat)
[![ModelScope link](https://img.shields.io/badge/ModelScope-Model-blue)](https://www.modelscope.cn/models/iic/ACE-0.6B-512px) [![HuggingFace link](https://img.shields.io/badge/%F0%9F%A4%97%20Hugging%20Face-Model-yellow)](https://huggingface.co/scepter-studio/ACE-0.6B-512px) | ## 🖥️ SCEPTER Studio diff --git a/scepter/studio/chatbot/chatbot.py b/scepter/studio/chatbot/chatbot.py index 4f04400..57629d7 100644 --- a/scepter/studio/chatbot/chatbot.py +++ b/scepter/studio/chatbot/chatbot.py @@ -3,12 +3,14 @@ import argparse import base64 import copy +import csv import glob import io import os import random import re import string +import sys import threading import cv2 @@ -31,6 +33,8 @@ from scepter.studio.utils.env import init_env from .example import get_examples from .utils import load_image +csv.field_size_limit(sys.maxsize) + refresh_sty = '\U0001f504' # 🔄 clear_sty = '\U0001f5d1' # 🗑️ upload_sty = '\U0001f5bc' # 🖼️ @@ -640,8 +644,8 @@ class ChatBotUI(object): print(new_message) imgs = self.pipe( - input_image=edit_image, - input_mask=edit_image_mask, + image=edit_image, + mask=edit_image_mask, task=edit_task, prompt=[new_message] * len(edit_image) if edit_image is not None else [new_message], @@ -832,8 +836,8 @@ class ChatBotUI(object): img_num = len(edit_image) if edit_image is not None else 1 imgs = self.pipe( - input_image=edit_image, - input_mask=edit_image_mask, + image=edit_image, + mask=edit_image_mask, task=edit_task, prompt=[prompt] * img_num, negative_prompt=[''] * img_num, @@ -875,6 +879,7 @@ class ChatBotUI(object): self.example_ref_im1, self.text, self.seed ], outputs=[self.chatbot, self.text, self.gallery], + examples_per_page=4, run_on_click=True) ########################################