Update app.py

app.py

@@ -15,7 +15,7 @@ proc5=gr.Interface.load("models/dreamlike-art/dreamlike-diffusion-1.0")
 
 import random
 
-def add_random_noise(prompt, noise_level=0.1):
+def add_random_noise(prompt, noise_level=0.07):
     # Get the percentage of characters to add as noise
     percentage_noise = noise_level * 5
     # Get the number of characters to add as noise
@@ -37,13 +37,8 @@ def send_it1(inputs, noise_level, proc1=proc1):
         if not queue.empty():
             queue.queue.clear()
         queue_length_counter = 0
-    try:
-        output1 = proc1(prompt_with_noise, timeout=60) #set a timeout of 60 seconds
-    except gr.TimeoutError:
-        print("The model took too long to generate a response. Exiting.")
-        sys.exit()
+    output1 = proc1(prompt_with_noise)
     queue_length_counter += 1
-    time.sleep(3)
     return output1
     time.sleep(1)
 
@@ -56,7 +51,6 @@ def send_it2(inputs, noise_level, proc1=proc1):
         queue_length_counter = 0
     output2 = proc1(prompt_with_noise)
     queue_length_counter += 1
-    time.sleep(3)
     return output2
     time.sleep(1)
 
@@ -69,7 +63,6 @@ def send_it3(inputs, noise_level, proc1=proc1):
         queue_length_counter = 0
     output3 = proc1(prompt_with_noise)
     queue_length_counter += 1
-    time.sleep(3)
     return output3
     time.sleep(1)
 
@@ -82,7 +75,6 @@ def send_it4(inputs, noise_level, proc1=proc1):
         queue_length_counter = 0
     output4 = proc1(prompt_with_noise)
     queue_length_counter += 1
-    time.sleep(3)
     return output4
     time.sleep(1)
 
@@ -95,7 +87,6 @@ def send_it5(inputs, noise_level, proc5=proc5):
         queue_length_counter = 0
     output5 = proc5(prompt_with_noise)
     queue_length_counter += 1
-    time.sleep(3)
     return output5
     time.sleep(1)
 
@@ -108,7 +99,6 @@ def send_it6(inputs, noise_level, proc5=proc5):
         queue_length_counter = 0
     output6 = proc5(prompt_with_noise)
     queue_length_counter += 1
-    time.sleep(3)
     return output6
     time.sleep(1)
 
@@ -121,7 +111,6 @@ def send_it7(inputs, noise_level, proc5=proc5):
         queue_length_counter = 0
     output7 = proc5(prompt_with_noise)
     queue_length_counter += 1
-    time.sleep(3)
     return output7
     time.sleep(1)
 
@@ -134,7 +123,6 @@ def send_it8(inputs, noise_level, proc5=proc5):
         queue_length_counter = 0
     output8 = proc5(prompt_with_noise)
     queue_length_counter += 1
-    time.sleep(3)
     return output8
     time.sleep(1)
 
@@ -148,7 +136,6 @@ def get_prompts(prompt_text):
         queue_length_counter = 0
     output = text_gen(prompt_text)
     queue_length_counter += 1
-    time.sleep(3)
     return output
     time.sleep(1)