Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -24,7 +24,7 @@ if 'past' not in st.session_state:
|
|
24 |
st.session_state['past'] = []
|
25 |
|
26 |
def query(payload):
|
27 |
-
bot_input_ids = tokenizer.encode(payload["text"] + tokenizer.eos_token, return_tensors='pt')
|
28 |
|
29 |
chat_history_ids = model.generate(
|
30 |
bot_input_ids, max_length=100,
|
|
|
24 |
st.session_state['past'] = []
|
25 |
|
26 |
def query(payload):
|
27 |
+
bot_input_ids = tokenizer.encode(payload["inputs"]["text"] + tokenizer.eos_token, return_tensors='pt')
|
28 |
|
29 |
chat_history_ids = model.generate(
|
30 |
bot_input_ids, max_length=100,
|