Spaces:
Sleeping
Sleeping
Update generate_answer.py
Browse files- 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-
|
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-
|
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 |
|