requirements2

app.py

@@ -5,11 +5,8 @@ from torchvision.utils import save_image
 import numpy as np
 from diffusers import StableDiffusionUpscalePipeline
 from huggingface_hub import hf_hub_download
-import torch
-
-# Load the model from Hugging Face Hub
-model_path = hf_hub_download(repo_id="Shaoan/ConceptGAN", filename="augceleba_6451.pkl")
 
+model_path = hf_hub_download(repo_id="Shaoan/ConceptGAN", filename="augceleba_8064.pkl")
 with open(model_path, 'rb') as f:
     G = pickle.load(f)['G_ema'].cpu().float()  # torch.nn.Module
 
@@ -138,30 +135,29 @@ def generate(seed, upscale, upscale_steps,*checkboxes):
 # Create the interface using gr.Blocks
 with gr.Blocks() as demo:
     with gr.Row():
-        sliders = [
-            gr.Slider(label='Bald', minimum=0, maximum=1, step=0.01),
-            gr.Slider(label='Black Hair', minimum=0, maximum=1, step=0.01),
-            gr.Slider(label='Blond Hair', minimum=0, maximum=1, step=0.01),
-            gr.Slider(label='Straight Hair', minimum=0, maximum=1, step=0.01),
-            gr.Slider(label='Wavy Hair', minimum=0, maximum=1, step=0.01),
-        ]
+        slider1 = gr.Slider(label='Not Bald          <-------------->  Bald', minimum=0, maximum=1, step=0.01)
+        slider2 = gr.Slider(label='No Black Hair     <-------->  Black Hair', minimum=0, maximum=1, step=0.01)
+        slider3 = gr.Slider(label='No Blond Hair     <-------->  Blond Hair', minimum=0, maximum=1, step=0.01)
+        slider4 = gr.Slider(label='No Straight Hair  <----->  Straight Hair', minimum=0, maximum=1, step=0.01)
+        slider5 = gr.Slider(label='No Wavy Hair      <------->  Wavy Hair', minimum=0, maximum=1, step=0.01)
+        sliders = [ slider1, slider2, slider3, slider4, slider5]
 
     with gr.Row():
-        sliders += [gr.Slider(label='Young', minimum=0, maximum=1, step=0.01)]
-        sliders += [gr.Slider(label='Male', minimum=0, maximum=1, step=0.01)]
+        sliders += [gr.Slider(label='Old            <--------------> Young', minimum=0, maximum=1, step=0.01)]
+        sliders += [gr.Slider(label='Female         <--------------> Male', minimum=0, maximum=1, step=0.01)]
 
     with gr.Row():
-        sliders += [gr.Slider(label='Goatee', minimum=0, maximum=1, step=0.01)]
-        sliders += [gr.Slider(label='Mustache', minimum=0, maximum=1, step=0.01)]
+        sliders += [gr.Slider(label='No Goatee         <--------------> Goatee', minimum=0, maximum=1, step=0.01)]
+        sliders += [gr.Slider(label='No Mustache       <--------------> Mustache', minimum=0, maximum=1, step=0.01)]
 
     with gr.Row():
         sliders += [
-            gr.Slider(label='Big Nose', minimum=0, maximum=1, step=0.01),
-            gr.Slider(label='Chubby', minimum=0, maximum=1, step=0.01),
-            gr.Slider(label='Eyeglasses', minimum=0, maximum=1, step=0.01),
-            gr.Slider(label='Pale Skin', minimum=0, maximum=1, step=0.01),
-            gr.Slider(label='Smiling', minimum=0, maximum=1, step=0.01),
-            gr.Slider(label='Wearing Hat', minimum=0, maximum=1, step=0.01),
+            gr.Slider(label='Small Nose     <-------> Big Nose', minimum=0, maximum=1, step=0.01),
+            gr.Slider(label='Slim     <-------->  Chubby', minimum=0, maximum=1, step=0.01),
+            gr.Slider(label='No Eyeglasses     <--------> Eyeglasses', minimum=0, maximum=1, step=0.01),
+            gr.Slider(label='Tan Skin     <-------> Pale Skin', minimum=0, maximum=1, step=0.01),
+            gr.Slider(label='Not Smiling      <---------> Smiling', minimum=0, maximum=1, step=0.01),
+            gr.Slider(label='No Hat   <---------> Wearing Hat', minimum=0, maximum=1, step=0.01),
         ]
 
     seed_input = gr.Number(label="Seed", value=6)
@@ -172,9 +168,11 @@ with gr.Blocks() as demo:
     generate_button = gr.Button("Generate")
 
     output_image = gr.Image(label="Generated Image")
+    for slider in sliders:
+        slider.change(fn=generate, inputs=[seed_input] + upscale_funcs + sliders, outputs=output_image)
 
     # Set the action for the button
     generate_button.click(fn=generate, inputs=[seed_input] + upscale_funcs +sliders, outputs=output_image)
 
 # Launch the demo
-demo.launch()
+demo.launch()

requirements.txt

@@ -8,5 +8,5 @@ psutil~=6.0.0
 gradio==4.36.1
 diffusers==0.20.0
 huggingface_hub==0.24.6
-transformers==4.19.2
+transformers==4.31.0
 accelerate