Update app.py

app.py

@@ -20,8 +20,8 @@ def warmer():
             end_time = time.time()
             duration = end_time - start_time
             print("Time elapsed for Pygmalion model : ", duration)
-            #if duration > 120: # Check if duration to generate tokens exceeds 2 minutes
-            restart()
+            if duration > 120: # Check if duration to generate tokens exceeds 2 minutes
+                restart()
                 
             start_time = time.time()
             text_output_2,image_output_2 = client.predict(