Update app.py

app.py

@@ -7,6 +7,7 @@ import string
 import time
 from queue import Queue
 from threading import Thread
+from concurrent.futures import ThreadPoolExecutor
 
 text_gen=gr.Interface.load("spaces/Omnibus/MagicPrompt-Stable-Diffusion")
 def get_prompts(prompt_text):
@@ -44,7 +45,8 @@ def send_it1(inputs, noise_level, proc1=proc1):
     while queue.qsize() >= queue_threshold:
         time.sleep(2)
     queue.put(prompt_with_noise)
-    output1 = proc1(prompt_with_noise)
+    with ThreadPoolExecutor() as executor:
+        output1 = executor.submit(proc1, prompt_with_noise).result()
     return output1
 
 def send_it2(inputs, noise_level, proc1=proc1):
@@ -52,7 +54,8 @@ def send_it2(inputs, noise_level, proc1=proc1):
     while queue.qsize() >= queue_threshold:
         time.sleep(2)
     queue.put(prompt_with_noise)
-    output2 = proc1(prompt_with_noise)
+    with ThreadPoolExecutor() as executor:
+        output2 = executor.submit(proc1, prompt_with_noise).result()
     return output2
 
 def send_it3(inputs, noise_level, proc1=proc1):