Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
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=
|
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}]
|