ravi6k commited on
Commit
5bdf46a
·
verified ·
1 Parent(s): 8e9c9e5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
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=embedding_model_name,
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
  )