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 takes
|
| 15 |
# Initial system message to set the behavior of the assistant
|
| 16 |
messages = [{"role": "system", "content": system_message}]
|
| 17 |
|
|
@@ -64,7 +64,7 @@ def find_relevant_segment(user_query, segments):
|
|
| 64 |
|
| 65 |
def generate_response(user_query, relevant_segment):
|
| 66 |
"""
|
| 67 |
-
Generate a response emphasizing the bot's capability in
|
| 68 |
"""
|
| 69 |
try:
|
| 70 |
user_message = f"Here is a local restaurant based on your information: {relevant_segment}"
|
|
|
|
| 11 |
|
| 12 |
openai.api_key = os.environ["OPENAI_API_KEY"]
|
| 13 |
|
| 14 |
+
system_message = "You are a restaurant recommending chatbot that takes details about a restaurant including type of restaurant, dietary restrictions, and budget and chooses a restaurant in Seattle which fits all of the user's criteria. Then you output the restaurant name and website link."
|
| 15 |
# Initial system message to set the behavior of the assistant
|
| 16 |
messages = [{"role": "system", "content": system_message}]
|
| 17 |
|
|
|
|
| 64 |
|
| 65 |
def generate_response(user_query, relevant_segment):
|
| 66 |
"""
|
| 67 |
+
Generate a response emphasizing the bot's capability in suggesting a restaurant.
|
| 68 |
"""
|
| 69 |
try:
|
| 70 |
user_message = f"Here is a local restaurant based on your information: {relevant_segment}"
|