Update app.py
Browse files
app.py
CHANGED
|
@@ -43,7 +43,7 @@ def get_seed(seed):
|
|
| 43 |
return random.randint(0, MAX_SEED)
|
| 44 |
|
| 45 |
@spaces.GPU(duration=30)
|
| 46 |
-
def generate(input=DEFAULT_INPUT, negative_input=DEFAULT_NEGATIVE_INPUT, height=DEFAULT_HEIGHT, width=DEFAULT_WIDTH, steps=1, guidance=0, seed=None):
|
| 47 |
|
| 48 |
print(input, negative_input, height, width, steps, guidance, seed)
|
| 49 |
model.to(DEVICE)
|
|
|
|
| 43 |
return random.randint(0, MAX_SEED)
|
| 44 |
|
| 45 |
@spaces.GPU(duration=30)
|
| 46 |
+
def generate(input=DEFAULT_INPUT, negative_input=DEFAULT_NEGATIVE_INPUT, height=DEFAULT_HEIGHT, width=DEFAULT_WIDTH, steps=1, guidance=0, seed=None, progress=gr.Progress(track_tqdm=True)):
|
| 47 |
|
| 48 |
print(input, negative_input, height, width, steps, guidance, seed)
|
| 49 |
model.to(DEVICE)
|