Spaces:
vilarin
/
Running on Zero

vilarin commited on
Commit
ec35e66
1 Parent(s): 8b1e96d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -0
app.py CHANGED
@@ -15,6 +15,12 @@ checkpoints = {
15
  }
16
  loaded = None
17
 
 
 
 
 
 
 
18
  # Ensure model and scheduler are initialized in GPU-enabled function
19
  if torch.cuda.is_available():
20
  pipe = DiffusionPipeline.from_pretrained(base, unet=unet, torch_dtype=torch.float16, variant="fp16").to("cuda")
 
15
  }
16
  loaded = None
17
 
18
+ CSS = """
19
+ .gradio-container {
20
+ max-width: 690px !important;
21
+ }
22
+ """
23
+
24
  # Ensure model and scheduler are initialized in GPU-enabled function
25
  if torch.cuda.is_available():
26
  pipe = DiffusionPipeline.from_pretrained(base, unet=unet, torch_dtype=torch.float16, variant="fp16").to("cuda")