Update app.py

app.py

@@ -37,7 +37,11 @@ def send_it1(inputs, noise_level, proc1=proc1):
         if not queue.empty():
             queue.queue.clear()
         queue_length_counter = 0
-    output1 = proc1(prompt_with_noise)
+    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()
     queue_length_counter += 1
     time.sleep(3)
     return output1