Update app.py

app.py

@@ -43,7 +43,12 @@ from langchain_core.messages import AIMessage, HumanMessage
 from langchain_core.output_parsers import StrOutputParser
 from langchain_core.runnables import RunnableBranch, RunnableLambda, RunnableParallel, RunnablePassthrough
 
-# Pinecone setup
+
+# Set environment variables for CUDA
+os.environ['PYTORCH_USE_CUDA_DSA'] = '1'
+os.environ['CUDA_LAUNCH_BLOCKING'] = '1'
+
+
 hf_token = os.getenv("HF_TOKEN")
 if hf_token is None:
     print("Please set your Hugging Face token in the environment variables.")
@@ -52,8 +57,11 @@ else:
 
 logging.basicConfig(level=logging.DEBUG)
 
+
 embeddings = OpenAIEmbeddings(api_key=os.environ['OPENAI_API_KEY'])
 
+
+# Pinecone setup
 from pinecone import Pinecone
 pc = Pinecone(api_key=os.environ['PINECONE_API_KEY'])