ThongCoding commited on
Commit
3190ec7
·
verified ·
1 Parent(s): ed7d958

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -21,8 +21,9 @@ def respond(message, history):
21
 
22
  # encode + generate
23
  inputs = tokenizer(prompt, return_tensors="pt")
 
24
  outputs = model.generate(
25
- inputs[0],
26
  max_new_tokens=200,
27
  do_sample=False, # greedy
28
  pad_token_id=tokenizer.eos_token_id
 
21
 
22
  # encode + generate
23
  inputs = tokenizer(prompt, return_tensors="pt")
24
+ print(**inputs)
25
  outputs = model.generate(
26
+ **inputs,
27
  max_new_tokens=200,
28
  do_sample=False, # greedy
29
  pad_token_id=tokenizer.eos_token_id