wenjiao commited on
Commit
f8df59a
1 Parent(s): 9b0414e
Files changed (1) hide show
  1. app.py +5 -2
app.py CHANGED
@@ -33,10 +33,13 @@ def displayTextBox():
33
 
34
  def set_msg():
35
  global queue_size
 
 
 
36
  if queue_size > int(os.environ["max_queue_size"]):
37
- return "The current traffic is high with " + str(queue_size) + " in the queue. Please wait a moment."
38
  else:
39
- return "The current traffic is not high. You can submit your job now."
40
 
41
 
42
  def img2img_generate(source_img, prompt, steps=25, strength=0.75, seed=42, guidance_scale=7.5):
 
33
 
34
  def set_msg():
35
  global queue_size
36
+ que_high_msg = "The current traffic is high with " + str(queue_size) + " in the queue. Please wait a moment."
37
+ que_normal_msg = "The current traffic is not high. You can submit your job now."
38
+
39
  if queue_size > int(os.environ["max_queue_size"]):
40
+ return que_high_msg
41
  else:
42
+ return que_normal_msg
43
 
44
 
45
  def img2img_generate(source_img, prompt, steps=25, strength=0.75, seed=42, guidance_scale=7.5):