RoAr777 commited on
Commit
5042e4f
·
1 Parent(s): 5601e25

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -15,7 +15,7 @@ def chatbot_model(m,history):
15
  outputs = model.generate(input_ids, max_length=1000, pad_token_id=tokenizer.eos_token_id)
16
 
17
  # Decode the output
18
- decoded_output = tokenizer.decode(outputs, skip_special_tokens=True)
19
  return decoded_output
20
 
21
  iface = gr.ChatInterface(chatbot_model)
 
15
  outputs = model.generate(input_ids, max_length=1000, pad_token_id=tokenizer.eos_token_id)
16
 
17
  # Decode the output
18
+ decoded_output = tokenizer.decode(outputs[0], skip_special_tokens=True)
19
  return decoded_output
20
 
21
  iface = gr.ChatInterface(chatbot_model)