Di Zhang
commited on
Update app.py
Browse files
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
|
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)]
|