m7n commited on
Commit
8f3219d
·
1 Parent(s): 6eed02d

Increase GPU duration for prediction function

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -22,7 +22,7 @@ app = FastAPI()
22
  # Mount the static directory
23
  app.mount("/static", StaticFiles(directory="static"), name="static")
24
 
25
- @spaces.GPU(duration=10)
26
  def predict(request: gr.Request, text_input):
27
  token = _get_token(request)
28
  file_name = f"{datetime.utcnow().strftime('%s')}.html"
 
22
  # Mount the static directory
23
  app.mount("/static", StaticFiles(directory="static"), name="static")
24
 
25
+ @spaces.GPU(duration=4*60)
26
  def predict(request: gr.Request, text_input):
27
  token = _get_token(request)
28
  file_name = f"{datetime.utcnow().strftime('%s')}.html"