Spaces:

karubiniumu
/

niwa-rag

Running

karubiniumu commited on 24 days ago

Commit

7068013

•

1 Parent(s): 46b2a15

gemini-1.5-flash

Files changed (1) hide show

app.py CHANGED Viewed

@@ -64,7 +64,7 @@ class Niwa_rag :
         document_joiner = DocumentJoiner()
         ranker = TransformersSimilarityRanker(model=ranker_model,top_k=8)
         prompt_builder = ChatPromptBuilder(template=messages,variables=["query", "documents", "memories"], required_variables=["query", "documents", "memories"])
-        gemini = GoogleAIGeminiChatGenerator(model="models/gemini-1.0-pro")
         memory_store = InMemoryChatMessageStore()
         memory_joiner = ListJoiner(List[ChatMessage])

         document_joiner = DocumentJoiner()
         ranker = TransformersSimilarityRanker(model=ranker_model,top_k=8)
         prompt_builder = ChatPromptBuilder(template=messages,variables=["query", "documents", "memories"], required_variables=["query", "documents", "memories"])
+        gemini = GoogleAIGeminiChatGenerator(model="models/gemini-1.5-flash")
         memory_store = InMemoryChatMessageStore()
         memory_joiner = ListJoiner(List[ChatMessage])