Update app.py
Browse files
app.py
CHANGED
@@ -111,7 +111,7 @@ def chat_gpt():
|
|
111 |
user_input = request.json.get('message')
|
112 |
if user_input is None:
|
113 |
return jsonify({'error': 'Message parameter missing'})
|
114 |
-
llm = LLM("
|
115 |
response = llm.chat([{"role": "user", "content": user_input}]) # Call the mistral_chat method
|
116 |
return jsonify({"response": response})
|
117 |
|
|
|
111 |
user_input = request.json.get('message')
|
112 |
if user_input is None:
|
113 |
return jsonify({'error': 'Message parameter missing'})
|
114 |
+
llm = LLM("meta-llama/Meta-Llama-3-70B-Instruct") # Initialize the LLM class with a model
|
115 |
response = llm.chat([{"role": "user", "content": user_input}]) # Call the mistral_chat method
|
116 |
return jsonify({"response": response})
|
117 |
|