ifire commited on
Commit
4a61237
·
verified ·
1 Parent(s): ac9fb9d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -25,7 +25,7 @@ def delete_file_after_ten_minutes(filename):
25
  time.sleep(600) # Wait for 10 minutes
26
  os.remove(filename)
27
 
28
- @spaces.GPU
29
  def predict(text, num_input, num_temp):
30
  transformer.eval()
31
  labels = [label.strip() for label in text.split(',')]
 
25
  time.sleep(600) # Wait for 10 minutes
26
  os.remove(filename)
27
 
28
+ @spaces.GPU(duration=120)
29
  def predict(text, num_input, num_temp):
30
  transformer.eval()
31
  labels = [label.strip() for label in text.split(',')]