JustusI commited on
Commit
b2daff9
·
verified ·
1 Parent(s): ab32ff3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -44,7 +44,7 @@ if prompt := st.chat_input():
44
 
45
  ##Get response to the message using client
46
  inputs = tokenizer.encode(text, add_special_tokens=False, return_tensors="pt")
47
- outputs = model.generate(**inputs, max_new_tokens=150)
48
 
49
 
50
 
 
44
 
45
  ##Get response to the message using client
46
  inputs = tokenizer.encode(text, add_special_tokens=False, return_tensors="pt")
47
+ outputs = model.generate(input_ids=inputs, max_new_tokens=150)
48
 
49
 
50