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

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(duration=120)
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()
29
  def predict(text, num_input, num_temp):
30
  transformer.eval()
31
  labels = [label.strip() for label in text.split(',')]