diff --git a/requirements.txt b/requirements.txt index 177fdde..e71044d 100644 --- a/requirements.txt +++ b/requirements.txt @@ -3,4 +3,7 @@ torchvision Ninja gradio huggingface_hub -hf_transfer \ No newline at end of file +hf_transfer +pyopengl +imgui +glfw diff --git a/scripts/gui.bat b/scripts/gui.bat new file mode 100644 index 0000000..d0ad247 --- /dev/null +++ b/scripts/gui.bat @@ -0,0 +1,12 @@ +@echo off +python visualizer_drag.py ^ + checkpoints/stylegan2_lions_512_pytorch.pkl ^ + checkpoints/stylegan2-ffhq-512x512.pkl ^ + checkpoints/stylegan2-afhqcat-512x512.pkl ^ + checkpoints/stylegan2-car-config-f.pkl ^ + checkpoints/stylegan2_dogs_1024_pytorch.pkl ^ + checkpoints/stylegan2_horses_256_pytorch.pkl ^ + checkpoints/stylegan2-cat-config-f.pkl ^ + checkpoints/stylegan2_elephants_512_pytorch.pkl ^ + checkpoints/stylegan_human_v2_512.pkl ^ + checkpoints/stylegan2-lhq-256x256.pkl