Update app.py

app.py

@@ -104,55 +104,50 @@ with block:
                 )
             btn = gr.Button("Generate image", scale=0)
 
-        # # sd_options = gr.Dropdown(options, label="SD options")
-        # sd_options = gr.Dropdown(options, value='SD1.4', label="SD options")
-        # # model_id = "CompVis/stable-diffusion-v1-4"
+        # sd_options = gr.Dropdown(options, label="SD options")
+        sd_options = gr.Dropdown(options, value='SD1.4', label="SD options")
+        # model_id = "CompVis/stable-diffusion-v1-4"
         
-        # if sd_options == 'SD1.4':
-        #     model_id = "CompVis/stable-diffusion-v1-4"
-        # elif sd_options == 'SD1.5':
-        #     model_id = "runwayml/stable-diffusion-v1-5"
-        # elif sd_options == 'SD2.1':
-        #     model_id = "stabilityai/stable-diffusion-2-1"
+        if sd_options == 'SD1.4':
+            model_id = "CompVis/stable-diffusion-v1-4"
+        elif sd_options == 'SD1.5':
+            model_id = "runwayml/stable-diffusion-v1-5"
+        elif sd_options == 'SD2.1':
+            model_id = "stabilityai/stable-diffusion-2-1"
             
-        # pip_sd = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16)
-        # pip_sd = pip_sd.to("cuda")
-
-        # pip_freeu = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16)
-        # pip_freeu = pip_freeu.to("cuda")
-        # # -------- freeu block registration
-        # register_free_upblock2d(pipe, b1=1.2, b2=1.4, s1=0.9, s2=0.2)
-        # register_free_crossattn_upblock2d(pipe, b1=1.2, b2=1.4, s1=0.9, s2=0.2)
-        # # -------- freeu block registration
+        pip_sd = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16)
+        pip_sd = pip_sd.to("cuda")
+
+        pip_freeu = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16)
+        pip_freeu = pip_freeu.to("cuda")
+        # -------- freeu block registration
+        register_free_upblock2d(pipe, b1=1.2, b2=1.4, s1=0.9, s2=0.2)
+        register_free_crossattn_upblock2d(pipe, b1=1.2, b2=1.4, s1=0.9, s2=0.2)
+        # -------- freeu block registration
         
-        # with gr.Accordion('FreeU Parameters', open=False):
+        with gr.Accordion('FreeU Parameters', open=False):
             
-        #     b1 = gr.Slider(label='b1: backbone factor of the first stage block of decoder',
-        #                             minimum=1,
-        #                             maximum=1.6,
-        #                             step=0.1,
-        #                             value=1)
-        #     b2 = gr.Slider(label='b2: backbone factor of the second stage block of decoder',
-        #                             minimum=1,
-        #                             maximum=1.6,
-        #                             step=0.1,
-        #                             value=1)
-        #     s1 = gr.Slider(label='s1: skip factor of the first stage block of decoder',
-        #                             minimum=0,
-        #                             maximum=1,
-        #                             step=0.1,
-        #                             value=1)
-        #     s2 = gr.Slider(label='s2: skip factor of the second stage block of decoder',
-        #                             minimum=0,
-        #                             maximum=1,
-        #                             step=0.1,
-        #                             value=1)
-        #     s3 = gr.Slider(label='s3: skip factor of the second stage block of decoder',
-        #                             minimum=0,
-        #                             maximum=1,
-        #                             step=0.1,
-        #                             value=1)
-
+            b1 = gr.Slider(label='b1: backbone factor of the first stage block of decoder',
+                                    minimum=1,
+                                    maximum=1.6,
+                                    step=0.1,
+                                    value=1)
+            b2 = gr.Slider(label='b2: backbone factor of the second stage block of decoder',
+                                    minimum=1,
+                                    maximum=1.6,
+                                    step=0.1,
+                                    value=1)
+            s1 = gr.Slider(label='s1: skip factor of the first stage block of decoder',
+                                    minimum=0,
+                                    maximum=1,
+                                    step=0.1,
+                                    value=1)
+            s2 = gr.Slider(label='s2: skip factor of the second stage block of decoder',
+                                    minimum=0,
+                                    maximum=1,
+                                    step=0.1,
+                                    value=1)
+        
         with gr.Row():
             with gr.Column(min_width=256) as c1:
                 image_1 = gr.Image(interactive=False)
@@ -162,11 +157,11 @@ with block:
                 image_2_label = gr.Markdown("FreeU")
             
 
-    ex = gr.Examples(examples=examples, fn=infer, inputs=[text], outputs=[image_1, image_2], cache_examples=False)
+    ex = gr.Examples(examples=examples, fn=infer, inputs=[text, pip_sd, pip_freeu], outputs=[image_1, image_2], cache_examples=False)
     ex.dataset.headers = [""]
 
     # text.submit(infer, inputs=[text, pip_sd, pip_freeu], outputs=[image_1, image_2])
     # btn.click(infer, inputs=[text, pip_sd, pip_freeu], outputs=[image_1, image_2])
 
-# block.launch()
-block.queue(default_enabled=False).launch(share=False)
+block.launch()
+# block.queue(default_enabled=False).launch(share=False)