Update app.py
Browse files
app.py
CHANGED
|
@@ -64,8 +64,8 @@ def infer(
|
|
| 64 |
width,
|
| 65 |
height,
|
| 66 |
num_inference_steps,
|
| 67 |
-
progress=gr.Progress(track_tqdm=True),
|
| 68 |
model='2k',
|
|
|
|
| 69 |
):
|
| 70 |
print("Using model:", model)
|
| 71 |
# if model == "2k":
|
|
@@ -190,12 +190,12 @@ with gr.Blocks(css=css) as demo:
|
|
| 190 |
fn=infer,
|
| 191 |
inputs=[
|
| 192 |
prompt,
|
| 193 |
-
model,
|
| 194 |
seed,
|
| 195 |
randomize_seed,
|
| 196 |
width,
|
| 197 |
height,
|
| 198 |
num_inference_steps,
|
|
|
|
| 199 |
],
|
| 200 |
outputs=[result, seed],
|
| 201 |
)
|
|
|
|
| 64 |
width,
|
| 65 |
height,
|
| 66 |
num_inference_steps,
|
|
|
|
| 67 |
model='2k',
|
| 68 |
+
progress=gr.Progress(track_tqdm=True),
|
| 69 |
):
|
| 70 |
print("Using model:", model)
|
| 71 |
# if model == "2k":
|
|
|
|
| 190 |
fn=infer,
|
| 191 |
inputs=[
|
| 192 |
prompt,
|
|
|
|
| 193 |
seed,
|
| 194 |
randomize_seed,
|
| 195 |
width,
|
| 196 |
height,
|
| 197 |
num_inference_steps,
|
| 198 |
+
model,
|
| 199 |
],
|
| 200 |
outputs=[result, seed],
|
| 201 |
)
|