Spaces:
Running
on
Zero
Running
on
Zero
johnsonhung906
commited on
Commit
·
66da5c9
1
Parent(s):
ef017a0
increase waiting time
Browse files
app.py
CHANGED
@@ -19,7 +19,7 @@ for name, config_path in model_configs.items():
|
|
19 |
models[name] = model_instance
|
20 |
detectors[name] = AttentionDetector(model_instance)
|
21 |
|
22 |
-
@spaces.GPU(duration=
|
23 |
def respond(message, threshold, model_name):
|
24 |
"""
|
25 |
Run the prompt injection detection and inference using the selected model.
|
|
|
19 |
models[name] = model_instance
|
20 |
detectors[name] = AttentionDetector(model_instance)
|
21 |
|
22 |
+
@spaces.GPU(duration=60)
|
23 |
def respond(message, threshold, model_name):
|
24 |
"""
|
25 |
Run the prompt injection detection and inference using the selected model.
|