Daemontatox commited on
Commit
7bf6caa
·
verified ·
1 Parent(s): 6816664

Update app.py

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