Spaces:
Sleeping
Sleeping
Increase GPU duration for prediction function
Browse files
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=
|
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"
|