Reverent commited on
Commit
46a75dc
·
1 Parent(s): 7e38619

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -18,7 +18,7 @@ def warmer():
18
  api_name="/predict"
19
  )
20
  end_time = time.time()
21
- duration = start_time - end_time
22
  print("Time elapsed for Pygmalion model : ", duration)
23
  if duration > 120: # Check if duration to generate tokens exceeds 2 minutes
24
  restart()
@@ -30,7 +30,7 @@ def warmer():
30
  api_name="/predict"
31
  )
32
  end_time = time.time()
33
- duration = start_time - end_time
34
  print("Time elapsed for WizardLM model : ", duration)
35
  if duration > 300: # Check if duration to generate tokens exceeds 5 minutes
36
  restart()
 
18
  api_name="/predict"
19
  )
20
  end_time = time.time()
21
+ duration = end_time - start_time
22
  print("Time elapsed for Pygmalion model : ", duration)
23
  if duration > 120: # Check if duration to generate tokens exceeds 2 minutes
24
  restart()
 
30
  api_name="/predict"
31
  )
32
  end_time = time.time()
33
+ duration = end_time - start_time
34
  print("Time elapsed for WizardLM model : ", duration)
35
  if duration > 300: # Check if duration to generate tokens exceeds 5 minutes
36
  restart()