Official repo for AdGPT: Explore Meaningful Advertising with ChatGPT
conda create -n AdGPT python=3.9
conda activate AdGPT
pip install -r requirements.txt
pip install torch==1.13.1+cu116 torchvision==0.14.1+cu116 --extra-index-url https://download.pytorch.org/whl/cu116
python -m pip install 'git+https://github.com/facebookresearch/detectron2.git'
Openai series
CUDA_VISIBLE_DEVICES=0 python demo.py --image_path ./img/test.jpg \
--openai \
--openai_key YOUR_API_KEY \
--openai_base https://api.openai.com/v1 \
--chat_model gpt-4o-mini
GLM series
CUDA_VISIBLE_DEVICES=0 python exp.py --image_path img/book.jpg \
--chatglm \
--glm_key YOUR_API_KEY \
--glm_model glm-4-flash \
--version cn \
Check OpenAI and GLM documents for how to get your own API.
python test.py --GSS
- Metrics-ChatGPT Rating
- Downstream Task-Slogan Generation
- Chinese Version AdGPT
We would like to express our gratitude to previous work, which includes but is not limited to: Diffusers, EasyOCR, Detic, Transformers, CNOCR and detectron2.