Update app.py
Browse files
app.py
CHANGED
@@ -20,7 +20,7 @@ pipe = pipe.to(device)
|
|
20 |
|
21 |
|
22 |
# Funzione per generare un'immagine
|
23 |
-
def generate_image(prompt, neg_prompt, width=resolution[0], height=resolution[1], seed, steps):
|
24 |
generator = torch.manual_seed(seed)
|
25 |
start_time = time.time()
|
26 |
image = pipe(prompt, height=height, width=width, num_inference_steps=steps, guidance_scale=guidance_scale, generator=generator, negative_prompt=neg_prompt).images[0]
|
|
|
20 |
|
21 |
|
22 |
# Funzione per generare un'immagine
|
23 |
+
def generate_image(prompt, neg_prompt, width=resolution[0], height=resolution[1], seed=1, steps=35):
|
24 |
generator = torch.manual_seed(seed)
|
25 |
start_time = time.time()
|
26 |
image = pipe(prompt, height=height, width=width, num_inference_steps=steps, guidance_scale=guidance_scale, generator=generator, negative_prompt=neg_prompt).images[0]
|