datboyalex commited on
Commit
b0c1ef9
·
verified ·
1 Parent(s): 759492f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -3
app.py CHANGED
@@ -62,18 +62,22 @@ vector_store = FAISS.from_texts(
62
  embedding=embeddings,
63
  metadatas=metadatas
64
  )
65
-
66
  # Step 5: Initialize the LLM
67
- openai_api_key = os.getenv("tauhid") # Securely access the API key from Hugging Face Secrets
68
  if not openai_api_key:
69
- raise ValueError("OPENAI_API_KEY environment variable is not set. Please add it to the Space Secrets.")
70
 
 
71
  llm = ChatOpenAI(
72
  model_name="gpt-4o-mini",
73
  temperature=0.7,
74
  api_key=openai_api_key
75
  )
76
 
 
 
 
 
77
  # Step 6: Set Up the RetrievalQA Chain
78
  retriever = vector_store.as_retriever(search_kwargs={"k": 5})
79
  qa_chain = RetrievalQA.from_chain_type(
 
62
  embedding=embeddings,
63
  metadatas=metadatas
64
  )
 
65
  # Step 5: Initialize the LLM
66
+ openai_api_key = os.getenv("tauhid") # Fetch the key using the actual secret name
67
  if not openai_api_key:
68
+ raise ValueError("tauhid environment variable is not set. Please add it to the Space Secrets.")
69
 
70
+ # Initialize the LLM
71
  llm = ChatOpenAI(
72
  model_name="gpt-4o-mini",
73
  temperature=0.7,
74
  api_key=openai_api_key
75
  )
76
 
77
+ # Initialize Embeddings with the same key
78
+ embeddings = OpenAIEmbeddings(openai_api_key=openai_api_key)
79
+
80
+
81
  # Step 6: Set Up the RetrievalQA Chain
82
  retriever = vector_store.as_retriever(search_kwargs={"k": 5})
83
  qa_chain = RetrievalQA.from_chain_type(