mgetz commited on
Commit
d8be181
·
verified ·
1 Parent(s): 45e6128

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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 details about a restaurant including type of restaurant, dietary restrictions, and budget and chooses a restaurant in Seattle which best fits 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
 
@@ -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-3.5-turbo",
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,