Update app.py

app.py

@@ -1,6 +1,4 @@
-"""
-Train a diffusion model on images.
-"""
+
 import gradio as gr
 import argparse
 from einops import rearrange