gunaEntrans commited on
Commit
ab5e782
β€’
1 Parent(s): 48e2e12

Rename chatbot.py to app.py

Browse files
Files changed (1) hide show
  1. 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