Allow number of images to be specified with environment variable

app.py

@@ -43,7 +43,6 @@ from gradio_scribble2image_interactive import \
 from gradio_seg2image import create_demo as create_demo_seg
 from model import Model
 
-MAX_IMAGES = 1
 DESCRIPTION = '''# ControlNet
 
 This is an unofficial demo for [https://github.com/lllyasviel/ControlNet](https://github.com/lllyasviel/ControlNet).
@@ -57,32 +56,54 @@ if (SPACE_ID := os.getenv('SPACE_ID')) is not None:
 <p/>
 '''
 
+MAX_IMAGES = int(os.getenv('MAX_IMAGES', '3'))
+DEFAULT_NUM_IMAGES = min(MAX_IMAGES, int(os.getenv('DEFAULT_NUM_IMAGES', '1')))
+
 model = Model()
 
 with gr.Blocks(css='style.css') as demo:
     gr.Markdown(DESCRIPTION)
     with gr.Tabs():
         with gr.TabItem('Canny'):
-            create_demo_canny(model.process_canny, max_images=MAX_IMAGES)
+            create_demo_canny(model.process_canny,
+                              max_images=MAX_IMAGES,
+                              default_num_images=DEFAULT_NUM_IMAGES)
         with gr.TabItem('Hough'):
-            create_demo_hough(model.process_hough, max_images=MAX_IMAGES)
+            create_demo_hough(model.process_hough,
+                              max_images=MAX_IMAGES,
+                              default_num_images=DEFAULT_NUM_IMAGES)
         with gr.TabItem('HED'):
-            create_demo_hed(model.process_hed, max_images=MAX_IMAGES)
+            create_demo_hed(model.process_hed,
+                            max_images=MAX_IMAGES,
+                            default_num_images=DEFAULT_NUM_IMAGES)
         with gr.TabItem('Scribble'):
-            create_demo_scribble(model.process_scribble, max_images=MAX_IMAGES)
+            create_demo_scribble(model.process_scribble,
+                                 max_images=MAX_IMAGES,
+                                 default_num_images=DEFAULT_NUM_IMAGES)
         with gr.TabItem('Scribble Interactive'):
             create_demo_scribble_interactive(
-                model.process_scribble_interactive, max_images=MAX_IMAGES)
+                model.process_scribble_interactive,
+                max_images=MAX_IMAGES,
+                default_num_images=DEFAULT_NUM_IMAGES)
         with gr.TabItem('Fake Scribble'):
             create_demo_fake_scribble(model.process_fake_scribble,
-                                      max_images=MAX_IMAGES)
+                                      max_images=MAX_IMAGES,
+                                      default_num_images=DEFAULT_NUM_IMAGES)
         with gr.TabItem('Pose'):
-            create_demo_pose(model.process_pose, max_images=MAX_IMAGES)
+            create_demo_pose(model.process_pose,
+                             max_images=MAX_IMAGES,
+                             default_num_images=DEFAULT_NUM_IMAGES)
         with gr.TabItem('Segmentation'):
-            create_demo_seg(model.process_seg, max_images=MAX_IMAGES)
+            create_demo_seg(model.process_seg,
+                            max_images=MAX_IMAGES,
+                            default_num_images=DEFAULT_NUM_IMAGES)
         with gr.TabItem('Depth'):
-            create_demo_depth(model.process_depth, max_images=MAX_IMAGES)
+            create_demo_depth(model.process_depth,
+                              max_images=MAX_IMAGES,
+                              default_num_images=DEFAULT_NUM_IMAGES)
         with gr.TabItem('Normal map'):
-            create_demo_normal(model.process_normal, max_images=MAX_IMAGES)
+            create_demo_normal(model.process_normal,
+                               max_images=MAX_IMAGES,
+                               default_num_images=DEFAULT_NUM_IMAGES)
 
 demo.queue(api_open=False).launch()

gradio_canny2image.py

@@ -3,7 +3,7 @@
 import gradio as gr
 
 
-def create_demo(process, max_images=12):
+def create_demo(process, max_images=12, default_num_images=3):
     with gr.Blocks() as demo:
         with gr.Row():
             gr.Markdown('## Control Stable Diffusion with Canny Edge Maps')
@@ -16,7 +16,7 @@ def create_demo(process, max_images=12):
                     num_samples = gr.Slider(label='Images',
                                             minimum=1,
                                             maximum=max_images,
-                                            value=1,
+                                            value=default_num_images,
                                             step=1)
                     image_resolution = gr.Slider(label='Image Resolution',
                                                  minimum=256,

gradio_depth2image.py

@@ -3,7 +3,7 @@
 import gradio as gr
 
 
-def create_demo(process, max_images=12):
+def create_demo(process, max_images=12, default_num_images=3):
     with gr.Blocks() as demo:
         with gr.Row():
             gr.Markdown('## Control Stable Diffusion with Depth Maps')
@@ -16,7 +16,7 @@ def create_demo(process, max_images=12):
                     num_samples = gr.Slider(label='Images',
                                             minimum=1,
                                             maximum=max_images,
-                                            value=1,
+                                            value=default_num_images,
                                             step=1)
                     image_resolution = gr.Slider(label='Image Resolution',
                                                  minimum=256,

gradio_fake_scribble2image.py

@@ -3,7 +3,7 @@
 import gradio as gr
 
 
-def create_demo(process, max_images=12):
+def create_demo(process, max_images=12, default_num_images=3):
     with gr.Blocks() as demo:
         with gr.Row():
             gr.Markdown('## Control Stable Diffusion with Fake Scribble Maps')
@@ -16,7 +16,7 @@ def create_demo(process, max_images=12):
                     num_samples = gr.Slider(label='Images',
                                             minimum=1,
                                             maximum=max_images,
-                                            value=1,
+                                            value=default_num_images,
                                             step=1)
                     image_resolution = gr.Slider(label='Image Resolution',
                                                  minimum=256,

gradio_hed2image.py

@@ -3,7 +3,7 @@
 import gradio as gr
 
 
-def create_demo(process, max_images=12):
+def create_demo(process, max_images=12, default_num_images=3):
     with gr.Blocks() as demo:
         with gr.Row():
             gr.Markdown('## Control Stable Diffusion with HED Maps')
@@ -16,7 +16,7 @@ def create_demo(process, max_images=12):
                     num_samples = gr.Slider(label='Images',
                                             minimum=1,
                                             maximum=max_images,
-                                            value=1,
+                                            value=default_num_images,
                                             step=1)
                     image_resolution = gr.Slider(label='Image Resolution',
                                                  minimum=256,

gradio_hough2image.py

@@ -3,7 +3,7 @@
 import gradio as gr
 
 
-def create_demo(process, max_images=12):
+def create_demo(process, max_images=12, default_num_images=3):
     with gr.Blocks() as demo:
         with gr.Row():
             gr.Markdown('## Control Stable Diffusion with Hough Line Maps')
@@ -16,7 +16,7 @@ def create_demo(process, max_images=12):
                     num_samples = gr.Slider(label='Images',
                                             minimum=1,
                                             maximum=max_images,
-                                            value=1,
+                                            value=default_num_images,
                                             step=1)
                     image_resolution = gr.Slider(label='Image Resolution',
                                                  minimum=256,

gradio_normal2image.py

@@ -3,7 +3,7 @@
 import gradio as gr
 
 
-def create_demo(process, max_images=12):
+def create_demo(process, max_images=12, default_num_images=3):
     with gr.Blocks() as demo:
         with gr.Row():
             gr.Markdown('## Control Stable Diffusion with Normal Maps')
@@ -16,7 +16,7 @@ def create_demo(process, max_images=12):
                     num_samples = gr.Slider(label='Images',
                                             minimum=1,
                                             maximum=max_images,
-                                            value=1,
+                                            value=default_num_images,
                                             step=1)
                     image_resolution = gr.Slider(label='Image Resolution',
                                                  minimum=256,

gradio_pose2image.py

@@ -3,7 +3,7 @@
 import gradio as gr
 
 
-def create_demo(process, max_images=12):
+def create_demo(process, max_images=12, default_num_images=3):
     with gr.Blocks() as demo:
         with gr.Row():
             gr.Markdown('## Control Stable Diffusion with Human Pose')
@@ -16,7 +16,7 @@ def create_demo(process, max_images=12):
                     num_samples = gr.Slider(label='Images',
                                             minimum=1,
                                             maximum=max_images,
-                                            value=1,
+                                            value=default_num_images,
                                             step=1)
                     image_resolution = gr.Slider(label='Image Resolution',
                                                  minimum=256,

gradio_scribble2image.py

@@ -3,7 +3,7 @@
 import gradio as gr
 
 
-def create_demo(process, max_images=12):
+def create_demo(process, max_images=12, default_num_images=3):
     with gr.Blocks() as demo:
         with gr.Row():
             gr.Markdown('## Control Stable Diffusion with Scribble Maps')
@@ -16,7 +16,7 @@ def create_demo(process, max_images=12):
                     num_samples = gr.Slider(label='Images',
                                             minimum=1,
                                             maximum=max_images,
-                                            value=1,
+                                            value=default_num_images,
                                             step=1)
                     image_resolution = gr.Slider(label='Image Resolution',
                                                  minimum=256,

gradio_scribble2image_interactive.py

@@ -8,7 +8,7 @@ def create_canvas(w, h):
     return np.zeros(shape=(h, w, 3), dtype=np.uint8) + 255
 
 
-def create_demo(process, max_images=12):
+def create_demo(process, max_images=12, default_num_images=3):
     with gr.Blocks() as demo:
         with gr.Row():
             gr.Markdown(
@@ -45,7 +45,7 @@ def create_demo(process, max_images=12):
                     num_samples = gr.Slider(label='Images',
                                             minimum=1,
                                             maximum=max_images,
-                                            value=1,
+                                            value=default_num_images,
                                             step=1)
                     image_resolution = gr.Slider(label='Image Resolution',
                                                  minimum=256,

gradio_seg2image.py

@@ -3,7 +3,7 @@
 import gradio as gr
 
 
-def create_demo(process, max_images=12):
+def create_demo(process, max_images=12, default_num_images=3):
     with gr.Blocks() as demo:
         with gr.Row():
             gr.Markdown('## Control Stable Diffusion with Segmentation Maps')
@@ -16,7 +16,7 @@ def create_demo(process, max_images=12):
                     num_samples = gr.Slider(label='Images',
                                             minimum=1,
                                             maximum=max_images,
-                                            value=1,
+                                            value=default_num_images,
                                             step=1)
                     image_resolution = gr.Slider(label='Image Resolution',
                                                  minimum=256,