mayzinoo commited on
Commit
2e7fddb
·
verified ·
1 Parent(s): ed4eea1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -3,7 +3,7 @@ import zipfile
3
  import gradio as gr
4
  from langchain_openai import ChatOpenAI
5
  from langchain.embeddings import HuggingFaceEmbeddings
6
- from langchain.vectorstores import Chroma
7
  from langchain.prompts import PromptTemplate
8
  from langchain.chains import LLMChain
9
 
@@ -18,7 +18,7 @@ vectordb = Chroma(persist_directory="geometry_chroma", embedding_function=embedd
18
  retriever = vectordb.as_retriever()
19
 
20
  # Set OpenAI key (use Secrets or .env later)
21
- os.environ["OPENAI_API_KEY"] = os.getenv("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")
22
 
23
  llm = ChatOpenAI(model_name="gpt-3.5-turbo", temperature=0.2)
24
 
 
3
  import gradio as gr
4
  from langchain_openai import ChatOpenAI
5
  from langchain.embeddings import HuggingFaceEmbeddings
6
+ from langchain_chroma import Chroma
7
  from langchain.prompts import PromptTemplate
8
  from langchain.chains import LLMChain
9
 
 
18
  retriever = vectordb.as_retriever()
19
 
20
  # Set OpenAI key (use Secrets or .env later)
21
+ os.environ["OPENAI_API_KEY"] = os.getenv("OPENAI_API_KEY")
22
 
23
  llm = ChatOpenAI(model_name="gpt-3.5-turbo", temperature=0.2)
24