next-chat-try / app.py
Pengyu-gis's picture
Create app.py
fd7f135
raw
history blame
266 Bytes
import os
# Set the CUDA_VISIBLE_DEVICES environment variable
os.environ["CUDA_VISIBLE_DEVICES"] = "0"
# Run the web demo
command = "python mllm/demo/web_demo.py --model_path AoZhang/nextchat-7b-336 --vit_path openai/clip-vit-large-patch14-336"
os.system(command)