Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
|
@@ -35,6 +35,7 @@ image_gen = StableDiffusion3Pipeline.from_pretrained(model_path, text_encoder_3=
|
|
| 35 |
|
| 36 |
image_gen = image_gen.to(device)
|
| 37 |
|
|
|
|
| 38 |
def generate_image(prompt, num_inference_steps=50, guidance_scale=7.5):
|
| 39 |
# Generate the image
|
| 40 |
result = image_gen(
|
|
|
|
| 35 |
|
| 36 |
image_gen = image_gen.to(device)
|
| 37 |
|
| 38 |
+
@spaces.GPU(enable_queue=True)
|
| 39 |
def generate_image(prompt, num_inference_steps=50, guidance_scale=7.5):
|
| 40 |
# Generate the image
|
| 41 |
result = image_gen(
|