shukdevdatta123 commited on
Commit
908fcb9
·
verified ·
1 Parent(s): f666f09

Update generate_answer.py

Browse files
Files changed (1) hide show
  1. generate_answer.py +2 -2
generate_answer.py CHANGED
@@ -26,7 +26,7 @@ def base_model_chatbot(messages):
26
  ]
27
  messages = system_message + messages
28
  response = openai.ChatCompletion.create(
29
- model="gpt-4o-mini", # Ensure the model is specified correctly
30
  messages=messages
31
  )
32
  return response.choices[0].message['content']
@@ -55,7 +55,7 @@ class VectorDB:
55
  class ConversationalRetrievalChain:
56
  """Class to manage the QA chain setup."""
57
 
58
- def __init__(self, model_name="gpt-4o-mini", temperature=0):
59
  self.model_name = model_name
60
  self.temperature = temperature
61
 
 
26
  ]
27
  messages = system_message + messages
28
  response = openai.ChatCompletion.create(
29
+ model="gpt-3.5-turbo", # Ensure the model is specified correctly
30
  messages=messages
31
  )
32
  return response.choices[0].message['content']
 
55
  class ConversationalRetrievalChain:
56
  """Class to manage the QA chain setup."""
57
 
58
+ def __init__(self, model_name="gpt-3.5-turbo", temperature=0):
59
  self.model_name = model_name
60
  self.temperature = temperature
61