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(
@@ -58,6 +58,7 @@ def restart():
     except Exception as e:
         print("NO INTERNET")
         print(str(e))
+    return
             
 def placeholder(dummy):
     return "lol"