Update app.py

app.py

@@ -104,29 +104,30 @@ with block:
                     container=False,
                 )
             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"
+        
+        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
         
         with gr.Accordion('FreeU Parameters', open=False):
-            # 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"
-                
-            # 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
-
             b1 = gr.Slider(label='b1: backbone factor of the first stage block of decoder',
                                     minimum=1,
                                     maximum=1.6,