Update app.py

app.py

@@ -1,36 +1,3 @@
-#   vae:
-#     class_path: src.models.vae.LatentVAE
-#     init_args:
-#       precompute: true
-#       weight_path: /mnt/bn/wangshuai6/models/sd-vae-ft-ema/
-#   denoiser:
-#     class_path: src.models.denoiser.decoupled_improved_dit.DDT
-#     init_args:
-#       in_channels: 4
-#       patch_size: 2
-#       num_groups: 16
-#       hidden_size: &hidden_dim 1152
-#       num_blocks: 28
-#       num_encoder_blocks: 22
-#       num_classes: 1000
-#   conditioner:
-#     class_path: src.models.conditioner.LabelConditioner
-#     init_args:
-#       null_class: 1000
-#   diffusion_sampler:
-#     class_path: src.diffusion.stateful_flow_matching.sampling.EulerSampler
-#     init_args:
-#       num_steps: 250
-#       guidance: 3.0
-#       state_refresh_rate: 1
-#       guidance_interval_min: 0.3
-#       guidance_interval_max: 1.0
-#       timeshift: 1.0
-#       last_step: 0.04
-#       scheduler: *scheduler
-#       w_scheduler: src.diffusion.stateful_flow_matching.scheduling.LinearScheduler
-#       guidance_fn: src.diffusion.base.guidance.simple_guidance_fn
-#       step_fn: src.diffusion.stateful_flow_matching.sampling.ode_step_fn
 import random
 import os
 import time
@@ -185,7 +152,7 @@ if __name__ == "__main__":
                 seed = gr.Slider(minimum=0, maximum=1000000, step=1, label="seed", value=0)
                 timeshift = gr.Slider(minimum=0.1, maximum=5.0, step=0.1, label="timeshift", value=3.0)
                 order = gr.Slider(minimum=1, maximum=4, step=1, label="order", value=2)
-            with gr.Column(scale=2):
+            with gr.Column(scale=1):
                 btn = gr.Button("Generate")
                 output_sample = gr.Image(label="Images")
             # with gr.Column(scale=2):