Spaces:
Yuanshi
/
Running on Zero

Yuanshi commited on
Commit
0f92491
·
verified ·
1 Parent(s): 0736b77

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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
  )