Jobin Thavu Varghese commited on
Commit ·
f706398
1
Parent(s): 885a344
gemini model
Browse files
config.py
CHANGED
|
@@ -9,7 +9,7 @@ HF_BUCKET_URL = os.environ.get("HF_BUCKET_URL")
|
|
| 9 |
HF_BUCKET_TOKEN = os.environ.get("HF_BUCKET_TOKEN")
|
| 10 |
|
| 11 |
model = ChatGoogleGenerativeAI(
|
| 12 |
-
model =
|
| 13 |
temperature = 1.0,
|
| 14 |
max_tokens = None,
|
| 15 |
timeout = 10,
|
|
|
|
| 9 |
HF_BUCKET_TOKEN = os.environ.get("HF_BUCKET_TOKEN")
|
| 10 |
|
| 11 |
model = ChatGoogleGenerativeAI(
|
| 12 |
+
model = os.getenv("GOOGLE_GEMINI_MODEL"),
|
| 13 |
temperature = 1.0,
|
| 14 |
max_tokens = None,
|
| 15 |
timeout = 10,
|