robinroy03 commited on
Commit
da60aad
1 Parent(s): 8bbaaaa
Files changed (1) hide show
  1. server.py +1 -1
server.py CHANGED
@@ -95,7 +95,7 @@ Example:
95
  Topic name: {message}
96
  Learner profile: {current_knowledge}
97
  """
98
- llm_model = google.generativeai.GenerativeModel('gemini-1.5-pro')
99
  chat_completion = llm_model.generate_content(sys_message)
100
  print(chat_completion)
101
  chat_completion_text = chat_completion.to_dict()['candidates'][0]['content']['parts'][0]['text']
 
95
  Topic name: {message}
96
  Learner profile: {current_knowledge}
97
  """
98
+ llm_model = google.generativeai.GenerativeModel('gemini-1.5-flash')
99
  chat_completion = llm_model.generate_content(sys_message)
100
  print(chat_completion)
101
  chat_completion_text = chat_completion.to_dict()['candidates'][0]['content']['parts'][0]['text']