GPU ONLY!

BEFORE LAUNCH TEST.PY, INSTALL THIS PACKAGES:
	pip install "git+https://github.com/ai-forever/Kandinsky-2.git"
	pip install gradio
	pip install numpy
	pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu117