phenomenon1981 commited on
Commit
94141d0
1 Parent(s): 0a6a0f2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +20 -28
app.py CHANGED
@@ -17,7 +17,7 @@ proc1=gr.Interface.load("models/dreamlike-art/dreamlike-photoreal-2.0")
17
  def reset_queue_periodically():
18
  start_time = time.time()
19
  while True:
20
- if time.time() - start_time > 300: # 300 seconds
21
  queue.queue.clear()
22
  start_time = time.time()
23
 
@@ -41,87 +41,79 @@ def add_random_noise(prompt, noise_level=0.07):
41
 
42
 
43
  def send_it1(inputs, noise_level, proc1=proc1):
44
- global queue
45
  prompt_with_noise = add_random_noise(inputs, noise_level)
46
- queue.put(prompt_with_noise)
47
- while queue.qsize() > queue_threshold:
48
  time.sleep(1)
 
49
  output1 = proc1(queue.get())
50
  return output1
51
 
52
  def send_it2(inputs, noise_level, proc1=proc1):
53
- global queue
54
  prompt_with_noise = add_random_noise(inputs, noise_level)
55
- queue.put(prompt_with_noise)
56
- while queue.qsize() > queue_threshold:
57
  time.sleep(1)
 
58
  output2 = proc1(queue.get())
59
  return output2
60
 
61
  def send_it3(inputs, noise_level, proc1=proc1):
62
- global queue
63
  prompt_with_noise = add_random_noise(inputs, noise_level)
64
- queue.put(prompt_with_noise)
65
- while queue.qsize() > queue_threshold:
66
  time.sleep(1)
 
67
  output3 = proc1(queue.get())
68
  return output3
69
 
70
  def send_it4(inputs, noise_level, proc1=proc1):
71
- global queue
72
  prompt_with_noise = add_random_noise(inputs, noise_level)
73
- queue.put(prompt_with_noise)
74
- while queue.qsize() > queue_threshold:
75
  time.sleep(1)
 
76
  output4 = proc1(queue.get())
77
  return output4
78
 
79
  def send_it5(inputs, noise_level, proc1=proc1):
80
- global queue
81
  prompt_with_noise = add_random_noise(inputs, noise_level)
82
- queue.put(prompt_with_noise)
83
- while queue.qsize() > queue_threshold:
84
  time.sleep(1)
 
85
  output5 = proc1(queue.get())
86
  return output5
87
 
88
  def send_it6(inputs, noise_level, proc1=proc1):
89
- global queue
90
  prompt_with_noise = add_random_noise(inputs, noise_level)
91
- queue.put(prompt_with_noise)
92
- while queue.qsize() > queue_threshold:
93
  time.sleep(1)
 
94
  output6 = proc1(queue.get())
95
  return output6
96
 
97
  def send_it7(inputs, noise_level, proc1=proc1):
98
- global queue
99
  prompt_with_noise = add_random_noise(inputs, noise_level)
100
- queue.put(prompt_with_noise)
101
- while queue.qsize() > queue_threshold:
102
  time.sleep(1)
 
103
  output7 = proc1(queue.get())
104
  return output7
105
 
106
  def send_it8(inputs, noise_level, proc1=proc1):
107
- global queue
108
  prompt_with_noise = add_random_noise(inputs, noise_level)
109
- queue.put(prompt_with_noise)
110
- while queue.qsize() > queue_threshold:
111
  time.sleep(1)
 
112
  output8 = proc1(queue.get())
113
  return output8
114
 
115
 
116
  def get_prompts(prompt_text):
117
- global queue
118
- queue.put(prompt_text)
119
- while queue.qsize() > queue_threshold:
120
  time.sleep(1)
 
121
  output = text_gen(queue.get())
122
  return output
123
 
124
 
 
125
  with gr.Blocks() as myface:
126
  with gr.Row():
127
 
 
17
  def reset_queue_periodically():
18
  start_time = time.time()
19
  while True:
20
+ if time.time() - start_time > 120: # 300 seconds
21
  queue.queue.clear()
22
  start_time = time.time()
23
 
 
41
 
42
 
43
  def send_it1(inputs, noise_level, proc1=proc1):
 
44
  prompt_with_noise = add_random_noise(inputs, noise_level)
45
+ while queue.qsize() >= queue_threshold:
 
46
  time.sleep(1)
47
+ queue.put(prompt_with_noise)
48
  output1 = proc1(queue.get())
49
  return output1
50
 
51
  def send_it2(inputs, noise_level, proc1=proc1):
 
52
  prompt_with_noise = add_random_noise(inputs, noise_level)
53
+ while queue.qsize() >= queue_threshold:
 
54
  time.sleep(1)
55
+ queue.put(prompt_with_noise)
56
  output2 = proc1(queue.get())
57
  return output2
58
 
59
  def send_it3(inputs, noise_level, proc1=proc1):
 
60
  prompt_with_noise = add_random_noise(inputs, noise_level)
61
+ while queue.qsize() >= queue_threshold:
 
62
  time.sleep(1)
63
+ queue.put(prompt_with_noise)
64
  output3 = proc1(queue.get())
65
  return output3
66
 
67
  def send_it4(inputs, noise_level, proc1=proc1):
 
68
  prompt_with_noise = add_random_noise(inputs, noise_level)
69
+ while queue.qsize() >= queue_threshold:
 
70
  time.sleep(1)
71
+ queue.put(prompt_with_noise)
72
  output4 = proc1(queue.get())
73
  return output4
74
 
75
  def send_it5(inputs, noise_level, proc1=proc1):
 
76
  prompt_with_noise = add_random_noise(inputs, noise_level)
77
+ while queue.qsize() >= queue_threshold:
 
78
  time.sleep(1)
79
+ queue.put(prompt_with_noise)
80
  output5 = proc1(queue.get())
81
  return output5
82
 
83
  def send_it6(inputs, noise_level, proc1=proc1):
 
84
  prompt_with_noise = add_random_noise(inputs, noise_level)
85
+ while queue.qsize() >= queue_threshold:
 
86
  time.sleep(1)
87
+ queue.put(prompt_with_noise)
88
  output6 = proc1(queue.get())
89
  return output6
90
 
91
  def send_it7(inputs, noise_level, proc1=proc1):
 
92
  prompt_with_noise = add_random_noise(inputs, noise_level)
93
+ while queue.qsize() >= queue_threshold:
 
94
  time.sleep(1)
95
+ queue.put(prompt_with_noise)
96
  output7 = proc1(queue.get())
97
  return output7
98
 
99
  def send_it8(inputs, noise_level, proc1=proc1):
 
100
  prompt_with_noise = add_random_noise(inputs, noise_level)
101
+ while queue.qsize() >= queue_threshold:
 
102
  time.sleep(1)
103
+ queue.put(prompt_with_noise)
104
  output8 = proc1(queue.get())
105
  return output8
106
 
107
 
108
  def get_prompts(prompt_text):
109
+ while queue.qsize() >= queue_threshold:
 
 
110
  time.sleep(1)
111
+ queue.put(prompt_text)
112
  output = text_gen(queue.get())
113
  return output
114
 
115
 
116
+
117
  with gr.Blocks() as myface:
118
  with gr.Row():
119