Update app.py
Browse files
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("
|
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 |
|