Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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
|
| 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("
|
| 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 |
|