Update app.py
Browse files
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 |
-
|
|
|
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 |
|