Abhaykoul commited on
Commit
e163cbd
1 Parent(s): f1d8af1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -132,7 +132,7 @@ def chat_gpt():
132
  if user_input is None:
133
  return jsonify({'error': 'Message parameter missing'})
134
 
135
- llm = LLM("mistralai/Mixtral-8x7B-Instruct-v0.1") # Initialize the LLM class with a model
136
  response = llm.mistral_chat([{"role": "user", "content": user_input}]) # Call the mistral_chat method
137
  return jsonify({"response": response})
138
 
 
132
  if user_input is None:
133
  return jsonify({'error': 'Message parameter missing'})
134
 
135
+ llm = LLM("databricks/dbrx-instruct") # Initialize the LLM class with a model
136
  response = llm.mistral_chat([{"role": "user", "content": user_input}]) # Call the mistral_chat method
137
  return jsonify({"response": response})
138