chrisjcc commited on
Commit
c2883ba
·
verified ·
1 Parent(s): cc7327b

Enable GPU access for processing

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -121,6 +121,7 @@ abot(next_prompt)
121
 
122
  action_re = re.compile('^Action: (\w+): (.*)$') # python regular expression to selection action
123
 
 
124
  def query(question, max_turns=5):
125
  i = 0
126
  bot = Agent(prompt)
 
121
 
122
  action_re = re.compile('^Action: (\w+): (.*)$') # python regular expression to selection action
123
 
124
+ @spaces.GPU(duration=120) # Designed to be effect-free in non-ZeroGPU environments, ensuring compatibility across different setups.
125
  def query(question, max_turns=5):
126
  i = 0
127
  bot = Agent(prompt)