Spaces:
Sleeping
Sleeping
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()
|