Spaces:
Paused
Paused
Add title and description to demo interface
Browse files
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()
|