Update app.py

app.py

@@ -45,38 +45,18 @@ def add_random_noise(prompt, noise_level=0.07):
 
 def send_it1(inputs, noise_level, proc1=proc1):
     prompt_with_noise = add_random_noise(inputs, noise_level)
-    while queue.qsize() >= queue_threshold:
-        time.sleep(2)
     queue.put(prompt_with_noise)
-    start_time = time.time()
-    while True:
-        try:
-            output1 = proc1(prompt_with_noise)
-            return output1
-        except:
-            if time.time() - start_time > 20:
-                return "Error: Image generation took too long. Please try again."
-            time.sleep(1)
+    output1 = proc1(prompt_with_noise)
+    return output1
 
 def send_it2(inputs, noise_level, proc1=proc1):
     prompt_with_noise = add_random_noise(inputs, noise_level)
-    while queue.qsize() >= queue_threshold:
-        time.sleep(2)
     queue.put(prompt_with_noise)
-    start_time = time.time()
-    while True:
-        try:
-            output2 = proc1(prompt_with_noise)
-            return output2
-        except:
-            if time.time() - start_time > 20:
-                return "Error: Image generation took too long. Please try again."
-            time.sleep(1)
+    output2 = proc1(prompt_with_noise)
+    return output2
 
 def send_it3(inputs, noise_level, proc1=proc1):
     prompt_with_noise = add_random_noise(inputs, noise_level)
-    while queue.qsize() >= queue_threshold:
-        time.sleep(2)
     queue.put(prompt_with_noise)
     output3 = proc1(prompt_with_noise)
     return output3