Sergidev commited on
Commit
42d891d
·
verified ·
1 Parent(s): be25071

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -15,7 +15,7 @@ model = None
15
  if tokenizer.pad_token is None:
16
  tokenizer.pad_token = tokenizer.eos_token
17
 
18
- @spaces.GPU
19
  def get_embedding(text):
20
  global model
21
  if model is None:
 
15
  if tokenizer.pad_token is None:
16
  tokenizer.pad_token = tokenizer.eos_token
17
 
18
+ @spaces.GPU(duration=300)
19
  def get_embedding(text):
20
  global model
21
  if model is None: