Di Zhang commited on
Commit
ddd8f10
·
verified ·
1 Parent(s): 5b033e9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -32,7 +32,7 @@ def llama_o1_template(data):
32
  text = template.format(content=data)
33
  return text
34
 
35
- @spaces.GPU(duration=200)
36
  def generate_text(message, history, max_tokens=512, temperature=0.9, top_p=0.95):
37
  temp = ""
38
  input_texts = [llama_o1_template(message)]
 
32
  text = template.format(content=data)
33
  return text
34
 
35
+ @spaces.GPU
36
  def generate_text(message, history, max_tokens=512, temperature=0.9, top_p=0.95):
37
  temp = ""
38
  input_texts = [llama_o1_template(message)]