vict0rsch commited on
Commit
0f77d9c
1 Parent(s): 5e61b3b

update readme

Browse files
Files changed (1) hide show
  1. README.md +25 -1
README.md CHANGED
@@ -37,7 +37,31 @@ If you use this code, data or pre-trained weights, please cite our ICLR 2022 pap
37
  }
38
  ```
39
 
40
- ## Using pre-trained weights
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
41
 
42
  In the paper, we present ClimateGAN as a solution to produce images of floods. It can actually do **more**:
43
 
 
37
  }
38
  ```
39
 
40
+ ## Using pre-trained weights from this Huggingface Space and Stable Diffusion In-painting
41
+
42
+ 1. Download code and model
43
+ ```
44
+ git clone https://huggingface.co/spaces/vict0rsch/climateGAN hf-spaces-climategan
45
+ cd hf-spaces-climategan
46
+ git lfs install
47
+ git lfs pull
48
+ ```
49
+ 2. Install requirements
50
+ ```
51
+ pip install requirements.txt
52
+ ```
53
+ 3. **Enable Stable Diffusion Inpainting** by visiting the model's card: https://huggingface.co/runwayml/stable-diffusion-inpainting **and** running `$ huggingface-cli login`
54
+ 4. Run `$ python climategan_wrapper.py help` for usage instructions on how to infer on a folder's images.
55
+ 5. Run `$ python app.py` to see the Gradio app.
56
+ 1. To use Google Street View you'll need an API key and set the `GMAPS_API_KEY` environment variable.
57
+ 2. To use Stable Diffusion if you can't run `$ huggingface-cli login` (on a Huggingface Space for instance) set the `HF_AUTH_TOKEN` env variable to a [Huggingface authorization token](https://huggingface.co/settings/tokens)
58
+ 3. To change the UI without model overhead, set the `CG_DEV_MODE` environment variable to `true`.
59
+
60
+ For a more fine-grained control on ClimateGAN's inferences, refer to `apply_events.py` (does not support Stable Diffusion painter)
61
+
62
+ **Note:** you don't have control on the prompt by design because I disabled the safety checker. Fork this space/repo and do it yourself if you really need to change the prompt. At least [open a discussion](https://huggingface.co/spaces/vict0rsch/climateGAN/discussions).
63
+
64
+ ## Using pre-trained weights from source
65
 
66
  In the paper, we present ClimateGAN as a solution to produce images of floods. It can actually do **more**:
67