merterbak commited on
Commit
dc6c5d6
·
verified ·
1 Parent(s): 16f671c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -24,7 +24,7 @@ def format_conversation_history(chat_history):
24
  messages.append({"role": role, "content": content})
25
  return messages
26
 
27
- @spaces.GPU(duration=90)
28
  def generate_response(input_data, chat_history, max_new_tokens, system_prompt, temperature, top_p, top_k, repetition_penalty):
29
  new_message = {"role": "user", "content": input_data}
30
  system_message = [{"role": "system", "content": system_prompt}] if system_prompt else []
 
24
  messages.append({"role": role, "content": content})
25
  return messages
26
 
27
+ @spaces.GPU(duration=60)
28
  def generate_response(input_data, chat_history, max_new_tokens, system_prompt, temperature, top_p, top_k, repetition_penalty):
29
  new_message = {"role": "user", "content": input_data}
30
  system_message = [{"role": "system", "content": system_prompt}] if system_prompt else []