Daemontatox commited on
Commit
eaca588
·
verified ·
1 Parent(s): 0f7b2cb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -97,7 +97,7 @@ def apply_llama3_chat_template(conversation, add_generation_prompt=True):
97
  prompt += "<|ASSISTANT|>\n"
98
  return prompt
99
 
100
- @spaces.GPU(duration=120)
101
  def generate_response(message, chat_history, system_prompt, temperature, max_tokens, top_p, top_k, repetition_penalty):
102
  # Build the conversation history.
103
  conversation = [{"role": "system", "content": system_prompt}]
 
97
  prompt += "<|ASSISTANT|>\n"
98
  return prompt
99
 
100
+ @spaces.GPU(duration=180)
101
  def generate_response(message, chat_history, system_prompt, temperature, max_tokens, top_p, top_k, repetition_penalty):
102
  # Build the conversation history.
103
  conversation = [{"role": "system", "content": system_prompt}]