Phoenix21 commited on
Commit
3bbc9f7
·
verified ·
1 Parent(s): b5475f0

Create models.py

Browse files
Files changed (1) hide show
  1. models.py +8 -0
models.py ADDED
@@ -0,0 +1,8 @@
 
 
 
 
 
 
 
 
 
1
+ # models.py
2
+ from google import genai
3
+ from langchain_groq import ChatGroq
4
+ from config import gemini_api_key, groq_api_key
5
+
6
+ client = genai.Client(api_key=gemini_api_key)
7
+ model_name = "gemini-2.0-flash-exp"
8
+ chat_model = ChatGroq(model="Gemma2-9b-It", groq_api_key=groq_api_key)