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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -124,12 +124,12 @@ def predict(user_input,company):
124
  # Get response from the LLM
125
  try:
126
  response = client.chat.completions.create(
127
- model=embedding_model,
128
  messages=prompt,
129
  temperature=0
130
  )
131
  print("responseRavi",response)
132
- prediction = response#.choices[0].message.content.strip()
133
  except Exception as e:
134
  prediction = f'Sorry, I encountered the following error: \n {e}'
135
 
 
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
  )
131
  print("responseRavi",response)
132
+ prediction = response.choices[0].message.content.strip()
133
  except Exception as e:
134
  prediction = f'Sorry, I encountered the following error: \n {e}'
135