Update app.py

app.py

@@ -389,12 +389,9 @@ def replay_diffusion(frames, replay_speed):
         time.sleep(delay)
 
 # Define the Gradio UI
-css = '''
-.gradio-container > .fillable {
-    max-width: 720px !important
-}
+css = '''.gradio-container > .fillable {max-width: 720px !important}
 '''
-with gr.Blocks(theme=gr.themes.Citrus()) as demo:
+with gr.Blocks(theme=gr.themes.Citrus(), css=css) as demo:
     gr.Markdown(
         """
         # The Annotated Discrete Diffusion Models