yxmauw commited on
Commit
9185ca3
·
verified ·
1 Parent(s): 844418d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -102,7 +102,8 @@ def generate_text(input_text, selected_model):
102
  # Define the chatbot function
103
  def generate_response(model_name, message, chat_history):
104
  model = load_model(model_name)
105
- response = model.generate(message, chat_history)
 
106
  chat_history.append((message, response))
107
  return chat_history, response
108
 
 
102
  # Define the chatbot function
103
  def generate_response(model_name, message, chat_history):
104
  model = load_model(model_name)
105
+ input_text = chat_history + " " + message
106
+ response = model.generate(input_text, max_tokens=100)
107
  chat_history.append((message, response))
108
  return chat_history, response
109