Update app.py

app.py

@@ -389,13 +389,16 @@ def replay_diffusion(frames, replay_speed):
         time.sleep(delay)
 
 # Define the Gradio UI
+css = '''
+.gradio-container > .fillable {
+    max-width: 720px !important
+}
+'''
 with gr.Blocks(theme=gr.themes.Citrus()) as demo:
     gr.Markdown(
         """
         # The Annotated Discrete Diffusion Models
-        
-        ## Tiny 7.23M parameters Shakespeare character diffusion model by [Ashwani Kumar](https://x.com/ash_at_tt/status/1977376958859092250)
-        
+        ### Tiny 7.23M parameters Shakespeare character diffusion model by [Ashwani Kumar](https://x.com/ash_at_tt/status/1977376958859092250)
         [GitHub](https://github.com/ash80/diffusion-gpt), [Colab](https://colab.research.google.com/github/ash80/diffusion-gpt/blob/master/The_Annotated_Discrete_Diffusion_Models.ipynb)
         """
     )