Update app.py
Browse files
app.py
CHANGED
@@ -36,7 +36,7 @@ model_gemini_pro = "gemini-pro"
|
|
36 |
|
37 |
|
38 |
# Configure Gemini models
|
39 |
-
Settings.llm = Gemini(
|
40 |
Settings.embed_model = GeminiEmbedding(
|
41 |
model_name="models/embedding-001",
|
42 |
api_key=os.getenv("GOOGLE_API_KEY")
|
|
|
36 |
|
37 |
|
38 |
# Configure Gemini models
|
39 |
+
Settings.llm = Gemini(models=model_gemini_pro, api_key=os.getenv("GOOGLE_API_KEY"))
|
40 |
Settings.embed_model = GeminiEmbedding(
|
41 |
model_name="models/embedding-001",
|
42 |
api_key=os.getenv("GOOGLE_API_KEY")
|