montehoover commited on
Commit
18d0655
·
verified ·
1 Parent(s): 2be6aff

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -135,7 +135,7 @@ class ModelWrapper:
135
  )
136
  return prompt
137
 
138
- @spaces.GPU(duration=240)
139
  def get_response(self, input, temperature=0.7, top_k=20, top_p=0.8, max_new_tokens=256,
140
  enable_thinking=True, system_prompt=SYSTEM_PROMPT):
141
  print("Generating response...")
 
135
  )
136
  return prompt
137
 
138
+ @spaces.GPU(duration=120)
139
  def get_response(self, input, temperature=0.7, top_k=20, top_p=0.8, max_new_tokens=256,
140
  enable_thinking=True, system_prompt=SYSTEM_PROMPT):
141
  print("Generating response...")