Spaces:

Trabis
/

RAG_loi

Running

Trabis commited on 3 days ago

Commit

0af70a9

verified ·

1 Parent(s): bc6bf26

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -226,7 +226,7 @@ class OptimizedRAGLoader:
 gemini_api_key = os.getenv("GEMINI_KEY")
 llm = ChatGoogleGenerativeAI(
-    model="Gemini 2.0 Flash-Lite",
     temperature=0,
     google_api_key=gemini_api_key,
     disable_streaming=True,

 gemini_api_key = os.getenv("GEMINI_KEY")
 llm = ChatGoogleGenerativeAI(
+    model="gemini-1.5-pro",
     temperature=0,
     google_api_key=gemini_api_key,
     disable_streaming=True,