Update app.py
Browse files
app.py
CHANGED
@@ -20,7 +20,7 @@ def warmer():
|
|
20 |
end_time = time.time()
|
21 |
duration = end_time - start_time
|
22 |
print("Time elapsed for Pygmalion model : ", duration)
|
23 |
-
if duration >
|
24 |
restart()
|
25 |
|
26 |
start_time = time.time()
|
|
|
20 |
end_time = time.time()
|
21 |
duration = end_time - start_time
|
22 |
print("Time elapsed for Pygmalion model : ", duration)
|
23 |
+
if duration > 90: # Check if duration to generate tokens exceeds 1 minutes
|
24 |
restart()
|
25 |
|
26 |
start_time = time.time()
|