Update app.py

app.py

@@ -8,7 +8,7 @@ import time
 from queue import Queue
 from threading import Thread
 
-text_gen=gr.Interface.load("spaces/Omnibus/MagicPrompt-Stable-Diffusion")
+text_gen=gr.Interface.load("spaces/Omnibus/MagicPrompt-Stable-Diffusion", live=True)
 proc1=gr.Interface.load("models/dreamlike-art/dreamlike-photoreal-2.0", live=True)
 
 def restart_script_periodically():
@@ -19,16 +19,6 @@ def restart_script_periodically():
 restart_thread = Thread(target=restart_script_periodically, daemon=True)
 restart_thread.start()
 
-def reset_queue_periodically():
-    start_time = time.time()
-    while True:
-        if time.time() - start_time >= 120: # 5 minutes
-            queue.queue.clear()
-            start_time = time.time()
-
-reset_queue_thread = Thread(target=reset_queue_periodically, daemon=True)
-reset_queue_thread.start()
-
 queue = Queue()
 queue_threshold = 9
 
@@ -53,7 +43,6 @@ def send_it1(inputs, noise_level, proc1=proc1):
         time.sleep(2)
     queue.put(prompt_with_noise)
     output1 = proc1(queue.get())
-    queue.queue.clear() # Clear the queue after the output is generated
     return output1
 
 def send_it2(inputs, noise_level, proc1=proc1):
@@ -62,7 +51,6 @@ def send_it2(inputs, noise_level, proc1=proc1):
         time.sleep(2)
     queue.put(prompt_with_noise)
     output2 = proc1(queue.get())
-    queue.queue.clear() # Clear the queue after the output is generated
     return output2
 
 def send_it3(inputs, noise_level, proc1=proc1):
@@ -71,7 +59,6 @@ def send_it3(inputs, noise_level, proc1=proc1):
         time.sleep(2)
     queue.put(prompt_with_noise)
     output3 = proc1(queue.get())
-    queue.queue.clear() # Clear the queue after the output is generated
     return output3
 
 def send_it4(inputs, noise_level, proc1=proc1):
@@ -80,7 +67,6 @@ def send_it4(inputs, noise_level, proc1=proc1):
         time.sleep(2)
     queue.put(prompt_with_noise)
     output4 = proc1(queue.get())
-    queue.queue.clear() # Clear the queue after the output is generated
     return output4
 
 def send_it5(inputs, noise_level, proc1=proc1):
@@ -89,7 +75,6 @@ def send_it5(inputs, noise_level, proc1=proc1):
         time.sleep(2)
     queue.put(prompt_with_noise)
     output5 = proc1(queue.get())
-    queue.queue.clear() # Clear the queue after the output is generated
     return output5
 
 def send_it6(inputs, noise_level, proc1=proc1):
@@ -98,7 +83,6 @@ def send_it6(inputs, noise_level, proc1=proc1):
         time.sleep(2)
     queue.put(prompt_with_noise)
     output6 = proc1(queue.get())
-    queue.queue.clear() # Clear the queue after the output is generated
     return output6
 
 def send_it7(inputs, noise_level, proc1=proc1):
@@ -107,7 +91,6 @@ def send_it7(inputs, noise_level, proc1=proc1):
         time.sleep(2)
     queue.put(prompt_with_noise)
     output7 = proc1(queue.get())
-    queue.queue.clear() # Clear the queue after the output is generated
     return output7
 
 def send_it8(inputs, noise_level, proc1=proc1):
@@ -116,7 +99,6 @@ def send_it8(inputs, noise_level, proc1=proc1):
         time.sleep(2)
     queue.put(prompt_with_noise)
     output8 = proc1(queue.get())
-    queue.queue.clear() # Clear the queue after the output is generated
     return output8
 
 
@@ -125,7 +107,6 @@ def get_prompts(prompt_text):
         time.sleep(2)
     queue.put(prompt_text)
     output = text_gen(queue.get())
-    queue.queue.clear() # Clear the queue after the output is generated
     return output
 
 
@@ -167,5 +148,4 @@ with gr.Blocks() as myface:
     see_prompts.click(get_prompts, inputs=[input_text], outputs=[prompt])
 
 myface.launch(enable_queue=True, inline=True)
-block.queue(concurrency_count=30, max_size=30)
-reset_queue_thread.join()
+block.queue(concurrency_count=30, max_size=30)