Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -122,9 +122,10 @@ def predict(user_input,company):
|
|
122 |
]
|
123 |
|
124 |
# Get response from the LLM
|
|
|
125 |
try:
|
126 |
response = client.chat.completions.create(
|
127 |
-
model=
|
128 |
messages=prompt,
|
129 |
temperature=0
|
130 |
)
|
|
|
122 |
]
|
123 |
|
124 |
# Get response from the LLM
|
125 |
+
model_name = 'mlabonne/NeuralHermes-2.5-Mistral-7B'
|
126 |
try:
|
127 |
response = client.chat.completions.create(
|
128 |
+
model=model_name,
|
129 |
messages=prompt,
|
130 |
temperature=0
|
131 |
)
|