SahaniJi commited on
Commit
1a5e302
·
verified ·
1 Parent(s): 002e937

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -14,7 +14,7 @@ MAX_SEED = np.iinfo(np.int32).max
14
  MAX_IMAGE_SIZE = 2048
15
 
16
  @spaces.GPU()
17
- def infer(prompt, seed=42, randomize_seed=False, width=1024, height=1024, num_inference_steps=4, progress=gr.Progress(track_tqdm=True)):
18
  if randomize_seed:
19
  seed = random.randint(0, MAX_SEED)
20
  generator = torch.Generator().manual_seed(seed)
@@ -119,4 +119,4 @@ with gr.Blocks(css=css) as demo:
119
  outputs = [result, seed]
120
  )
121
 
122
- demo.launch(enable_queue=True)
 
14
  MAX_IMAGE_SIZE = 2048
15
 
16
  @spaces.GPU()
17
+ def infer(prompt, seed=42, randomize_seed=False, width=1024, height=1024, num_inference_steps=4, progress=gr.Progress(track_tqdm=True),enable_queue=True):
18
  if randomize_seed:
19
  seed = random.randint(0, MAX_SEED)
20
  generator = torch.Generator().manual_seed(seed)
 
119
  outputs = [result, seed]
120
  )
121
 
122
+ demo.launch()