Abhaykoul commited on
Commit
c8cacfa
1 Parent(s): f180a2e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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("mistralai/Mixtral-8x7B-Instruct-v0.1") # 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
 
 
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