Update app.py

app.py

@@ -15,6 +15,12 @@ checkpoints = {
 }
 loaded = None
 
+CSS = """
+.gradio-container {
+  max-width: 690px !important;
+}
+"""
+
 # Ensure model and scheduler are initialized in GPU-enabled function
 if torch.cuda.is_available():
     pipe = DiffusionPipeline.from_pretrained(base, unet=unet, torch_dtype=torch.float16, variant="fp16").to("cuda")