simonzhang5429 commited on
Commit
fa1d1d8
·
verified ·
1 Parent(s): ac06820

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -18,7 +18,7 @@ def generate(text):
18
  text = text.strip()
19
  if text:
20
  chat_data.append({"role": "system", "content": text})
21
- outputs = model.generate(tokenizer.apply_chat_template(chat_data, return_tensors="pt"),
22
  max_new_tokens=128,
23
  top_p=0.01,
24
  top_k=85,
 
18
  text = text.strip()
19
  if text:
20
  chat_data.append({"role": "system", "content": text})
21
+ outputs = model.generate(tokenizer.apply_chat_template(chat_data),
22
  max_new_tokens=128,
23
  top_p=0.01,
24
  top_k=85,