thliang01 commited on
Commit
dfa7b37
·
unverified ·
1 Parent(s): eaeb551

Add title and description to demo interface

Browse files
Files changed (1) hide show
  1. app.py +3 -0
app.py CHANGED
@@ -10,6 +10,7 @@ weight_type = torch.float16
10
  pipe = StableDiffusionPipeline.from_pretrained(repo, torch_dtype=weight_type)
11
 
12
  generator = pipe
 
13
  # move to GPU if available
14
  if torch.cuda.is_available():
15
  generator = generator.to("cuda")
@@ -24,6 +25,8 @@ demo = gr.Interface(
24
  generate,
25
  "textbox",
26
  "image",
 
 
27
  batch=True,
28
  max_batch_size=4, # Set the batch size based on your CPU/GPU memory
29
  ).queue()
 
10
  pipe = StableDiffusionPipeline.from_pretrained(repo, torch_dtype=weight_type)
11
 
12
  generator = pipe
13
+
14
  # move to GPU if available
15
  if torch.cuda.is_available():
16
  generator = generator.to("cuda")
 
25
  generate,
26
  "textbox",
27
  "image",
28
+ title="SDXS: Real-Time One-Step Latent Diffusion Models with Image Conditions",
29
+ description="This demo showcases [SDXS](https://arxiv.org/abs/2403.16627)",
30
  batch=True,
31
  max_batch_size=4, # Set the batch size based on your CPU/GPU memory
32
  ).queue()