Update app.py
Browse files
app.py
CHANGED
|
@@ -11,7 +11,7 @@ retrieval_model_name = 'output/sentence-transformer-finetuned/'
|
|
| 11 |
|
| 12 |
openai.api_key = os.environ["OPENAI_API_KEY"]
|
| 13 |
|
| 14 |
-
system_message = "You are a restaurant recommending chatbot that
|
| 15 |
# Initial system message to set the behavior of the assistant
|
| 16 |
messages = [{"role": "system", "content": system_message}]
|
| 17 |
|
|
@@ -73,7 +73,7 @@ def generate_response(user_query, relevant_segment):
|
|
| 73 |
messages.append({"role": "user", "content": user_message})
|
| 74 |
|
| 75 |
response = openai.ChatCompletion.create(
|
| 76 |
-
model="gpt-
|
| 77 |
messages=messages,
|
| 78 |
max_tokens=150,
|
| 79 |
temperature=0.2,
|
|
|
|
| 11 |
|
| 12 |
openai.api_key = os.environ["OPENAI_API_KEY"]
|
| 13 |
|
| 14 |
+
system_message = "You are a restaurant recommending chatbot that suggests one restaurant based on the criteria the user provides."
|
| 15 |
# Initial system message to set the behavior of the assistant
|
| 16 |
messages = [{"role": "system", "content": system_message}]
|
| 17 |
|
|
|
|
| 73 |
messages.append({"role": "user", "content": user_message})
|
| 74 |
|
| 75 |
response = openai.ChatCompletion.create(
|
| 76 |
+
model="gpt-4o",
|
| 77 |
messages=messages,
|
| 78 |
max_tokens=150,
|
| 79 |
temperature=0.2,
|