Spaces:
Build error
Build error
Create models.py
Browse files
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)
|