Spaces:
Sleeping
Sleeping
Update generate_answer.py
Browse files- generate_answer.py +4 -3
generate_answer.py
CHANGED
@@ -22,12 +22,13 @@ openai.api_key = api_key
|
|
22 |
|
23 |
def base_model_chatbot(messages):
|
24 |
system_message = [
|
25 |
-
{"role": "system", "content": "You are
|
26 |
]
|
27 |
messages = system_message + messages
|
28 |
response = openai.ChatCompletion.create(
|
29 |
-
model="gpt-3.5-turbo",
|
30 |
-
messages=messages
|
|
|
31 |
)
|
32 |
return response.choices[0].message['content']
|
33 |
|
|
|
22 |
|
23 |
def base_model_chatbot(messages):
|
24 |
system_message = [
|
25 |
+
{"role": "system", "content": "You are a helpful AI chatbot, that answers questions asked by User."}
|
26 |
]
|
27 |
messages = system_message + messages
|
28 |
response = openai.ChatCompletion.create(
|
29 |
+
model="gpt-3.5-turbo",
|
30 |
+
messages=messages,
|
31 |
+
max_tokens=2048 # Increase max_tokens limit (or adjust based on your needs)
|
32 |
)
|
33 |
return response.choices[0].message['content']
|
34 |
|