Update app.py

app.py

@@ -20,7 +20,7 @@ def warmer():
             end_time = time.time()
             duration = end_time - start_time
             print("Time elapsed for Pygmalion model : ", duration)
-            if duration > 50: # Check if duration to generate tokens exceeds 2 minutes
+            if duration > 120: # Check if duration to generate tokens exceeds 2 minutes
                 restart()
                 
             start_time = time.time()
@@ -32,7 +32,7 @@ def warmer():
             end_time = time.time()
             duration = end_time - start_time
             print("Time elapsed for WizardLM model : ", duration)
-            if duration > 90: # Check if duration to generate tokens exceeds 5 minutes
+            if duration > 120: # Check if duration to generate tokens exceeds 5 minutes
                 restart()
                 
             if (text_output != None) and (text_output_2 != None):