Update image_generator.py

image_generator.py

@@ -89,7 +89,7 @@ def generate_images(
     python gen_images.py --outdir=out --trunc=0.7 --seeds=600-605 \\
         --network=https://api.ngc.nvidia.com/v2/models/nvidia/research/stylegan3/versions/1/files/stylegan3-t-metfacesu-1024x1024.pkl
     """
-    model_owner = "Cropinky"
+    model_owner = "huggan"
     #inputs = gr.inputs.Radio(["Abstract Expressionism", "Impressionism", "Cubism", "Minimalism", "Pop Art", "Color Field", "Hana Hanak houses"])
     model_path_dict = {
         'Impressionism' : 'projected_gan_impressionism',
@@ -97,8 +97,11 @@ def generate_images(
         'Abstract Expressionism' : 'projected_gan_abstract_expressionism', 
         'Pop Art' : 'projected_gan_pop_art',
         'Minimalism' : 'projected_gan_minimalism',
-        'Color Field' : 'Projected_gan_color_field',
-        'Hana Hanak houses' : 'projected_gana_hana' 
+        'Color Field' : 'projected_gan_color_field',
+        'Hana Hanak houses' : 'projected_gan_Hana_Hanak',
+        'Hana Hanak houses - abstract expressionism' : 'projected_gan_abstract_expressionism_hana',
+        'Hana Hanak houses - color field' : 'projected_gan_color_field_hana',
+        
     }
     
     model_path = model_owner + "/" + model_path_dict[model_path]