Update app.py

app.py

@@ -163,7 +163,7 @@ custom_css = """
 }
 """
 
-with gr.Blocks(css=custom_css) as demo:
+with gr.Blocks() as demo:
     gr.Markdown("# CIFAR-10 Diffusion Model")
     gr.Markdown("Select a class and click 'Generate' to create image samples.")
 
@@ -190,4 +190,4 @@ with gr.Blocks(css=custom_css) as demo:
 
 
 # Run the Gradio app
-demo.launch()
+demo.launch(css=custom_css)