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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -102,7 +102,10 @@ 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
- 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
 
102
  # Define the chatbot function
103
  def generate_response(model_name, message, chat_history):
104
  model = load_model(model_name)
105
+ if chat_history:
106
+ input_text = chat_history + " " + message
107
+ else:
108
+ input_text = message
109
  response = model.generate(input_text, max_tokens=100)
110
  chat_history.append((message, response))
111
  return chat_history, response