Update app.py

app.py

@@ -1,15 +1,8 @@
 import gradio as gr
 from PIL import Image  
 import torch
-from muse import PipelineMuse
-# from diffusers import AutoPipelineForText2Image, UniPCMultistepScheduler
-
-
-# import argparse, os, sys, glob
-# sys.path.append(os.path.split(sys.path[0])[0])
 
 from diffusers import StableDiffusionPipeline
-import torch
 from free_lunch_utils import register_free_upblock2d, register_free_crossattn_upblock2d
 
 torch.manual_seed(42)
@@ -114,8 +107,7 @@ with block:
                     placeholder="Enter your prompt",
                     container=False,
                 )
-            # btn = gr.Button("Generate image", scale=0)
-            btn = gr.Button("Generate image")
+            btn = gr.Button("Generate image", scale=0)
 
         # sd_options = gr.Dropdown(options, value='SD1.4', label="SD options")
         # model_id = "CompVis/stable-diffusion-v1-4"
@@ -137,28 +129,28 @@ with block:
         # register_free_crossattn_upblock2d(pip_freeu, 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)
+        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: