Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
|
@@ -36,7 +36,7 @@ print("Pipeline loaded!")
|
|
| 36 |
@spaces.GPU
|
| 37 |
def generate_image(prompt, height, width, num_inference_steps, seed=42, randomize_seed=True, progress=gr.Progress(track_tqdm=True)):
|
| 38 |
"""Generate an image from the given prompt."""
|
| 39 |
-
|
| 40 |
seed = random.randint(0, MAX_SEED)
|
| 41 |
image = pipe(
|
| 42 |
prompt=prompt,
|
|
|
|
| 36 |
@spaces.GPU
|
| 37 |
def generate_image(prompt, height, width, num_inference_steps, seed=42, randomize_seed=True, progress=gr.Progress(track_tqdm=True)):
|
| 38 |
"""Generate an image from the given prompt."""
|
| 39 |
+
if randomize_seed:
|
| 40 |
seed = random.randint(0, MAX_SEED)
|
| 41 |
image = pipe(
|
| 42 |
prompt=prompt,
|