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)]
|