Spaces:

Sergidev
/

3dembed

Running

Sergidev commited on Aug 20, 2024

Commit

42d891d

verified ·

1 Parent(s): be25071

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -15,7 +15,7 @@ model = None
 if tokenizer.pad_token is None:
     tokenizer.pad_token = tokenizer.eos_token
-@spaces.GPU
 def get_embedding(text):
     global model
     if model is None:

 if tokenizer.pad_token is None:
     tokenizer.pad_token = tokenizer.eos_token
+@spaces.GPU(duration=300)
 def get_embedding(text):
     global model
     if model is None: