change text + add cuda

models/model_settings.py

@@ -97,6 +97,6 @@ STYLEGAN_TRUNCATION_LAYERS = 8  # 0 means no truncation
 STYLEGAN_RANDOMIZE_NOISE = False
 
 # Settings for model running.
-USE_CUDA = False
+USE_CUDA = True
 
 MAX_IMAGES_ON_DEVICE = 8

utils/constants.py

@@ -9,7 +9,7 @@ VALID_CHOICES = [
     "Smiling",
     "Gray_Hair",
 ]
-ENABLE_GPU = False
+ENABLE_GPU = True
 MODEL_NAME = "stylegan_ffhq"
 OUTPUT_LIST = [
     gr.outputs.Image(type="pil", label="Generated Images"),
@@ -41,6 +41,7 @@ This is an interactive demo of an extension of the <a href="https://genforce.git
   <li>🔥 Submit!</li>
 </ul>
 Check the <a href="https://github.com/genforce/interfacegan">original repo</a> as well as the <a href="https://github.com/younesbelkada/interfacegan">extended version of the work</a>.
+&#x2B55; This method is biased on the data it has been trained for attribute recognition. E.g. if you decide to modify the "Bald" attribute on Female faces, the method will turn it into Male faces. Future work may focus more on this direction to try to have unbiased results of the modifications.
 
 </center>
 </p>