johnsonhung906 commited on
Commit
66da5c9
·
1 Parent(s): ef017a0

increase waiting time

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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=30)
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.