Spaces:

bla
/

tranny

Runtime error

Mbonea commited on Dec 13, 2023

Commit

6fe954e

1 Parent(s): 5e39b64

gemini-pro update

Files changed (1) hide show

App/Chat/utils/RAG.py CHANGED Viewed

@@ -8,7 +8,7 @@ PALM_API = ""
 API_KEY = os.environ.get("PALM_API", PALM_API)
 palm.configure(api_key=API_KEY)
 class GenerativeAIAssistant:
-    def __init__(self, api_key=API_KEY, model='chat-bison-001', temperature=0.85,
                  candidate_count=1, top_k=40, top_p=0.95):
         self.api_key = api_key
         self.model = model

 API_KEY = os.environ.get("PALM_API", PALM_API)
 palm.configure(api_key=API_KEY)
 class GenerativeAIAssistant:
+    def __init__(self, api_key=API_KEY, model='gemini-pro', temperature=0.85,
                  candidate_count=1, top_k=40, top_p=0.95):
         self.api_key = api_key
         self.model = model