Spaces:
Runtime error
Runtime error
gunaEntrans
commited on
Commit
β’
ab5e782
1
Parent(s):
48e2e12
Rename chatbot.py to app.py
Browse files- chatbot.py β app.py +4 -1
chatbot.py β app.py
RENAMED
@@ -9,11 +9,15 @@ from langchain_community.embeddings.sentence_transformer import (
|
|
9 |
from langchain_google_genai import ChatGoogleGenerativeAI
|
10 |
from langchain_community.vectorstores import Chroma
|
11 |
import gradio as gr
|
|
|
|
|
|
|
12 |
|
13 |
embedding = SentenceTransformerEmbeddings(
|
14 |
model_name="all-MiniLM-L6-v2"
|
15 |
)
|
16 |
|
|
|
17 |
# embedding = GoogleGenerativeAIEmbeddings()
|
18 |
db = Chroma(persist_directory="vectordb", embedding_function=embedding)
|
19 |
|
@@ -25,7 +29,6 @@ retriever_db = db.as_retriever()
|
|
25 |
# model = ChatOpenAI(model="gpt-3.5-turbo-1106",
|
26 |
# temperature=0, openai_api_key=opena_api)
|
27 |
|
28 |
-
api = "AIzaSyDHbc2aebeK5h5ZlxsZBr82ueAaOOUZgm8"
|
29 |
|
30 |
model = ChatGoogleGenerativeAI(
|
31 |
model="gemini-pro", google_api_key=api, temperature=0.1, stream=True
|
|
|
9 |
from langchain_google_genai import ChatGoogleGenerativeAI
|
10 |
from langchain_community.vectorstores import Chroma
|
11 |
import gradio as gr
|
12 |
+
import os
|
13 |
+
|
14 |
+
api = os.environ['google_api'].
|
15 |
|
16 |
embedding = SentenceTransformerEmbeddings(
|
17 |
model_name="all-MiniLM-L6-v2"
|
18 |
)
|
19 |
|
20 |
+
|
21 |
# embedding = GoogleGenerativeAIEmbeddings()
|
22 |
db = Chroma(persist_directory="vectordb", embedding_function=embedding)
|
23 |
|
|
|
29 |
# model = ChatOpenAI(model="gpt-3.5-turbo-1106",
|
30 |
# temperature=0, openai_api_key=opena_api)
|
31 |
|
|
|
32 |
|
33 |
model = ChatGoogleGenerativeAI(
|
34 |
model="gemini-pro", google_api_key=api, temperature=0.1, stream=True
|