Update app.py

app.py

@@ -1,14 +1,7 @@
 import subprocess
 
-# Define the command
-command = 'CMAKE_ARGS="-DLLAMA_CUBLAS=on" FORCE_CMAKE=1 pip install llama-cpp-python --no-cache-dir'
-
-# Run the command using subprocess
-try:
-    subprocess.run(command, shell=True, check=True)
-    print("Command executed successfully.")
-except subprocess.CalledProcessError as e:
-    print(f"Error: {e}")
+import os
+os.environ["TOKENIZERS_PARALLELISM"] = "false"
 
 import streamlit as st
 from langchain.llms import LlamaCpp
@@ -80,6 +73,17 @@ def init_retriever():
 
 # Check if retriever is already initialized in the session state
 if "retriever" not in st.session_state:
+    
+    # Define the command
+    command = 'CMAKE_ARGS="-DLLAMA_CUBLAS=on" FORCE_CMAKE=1 pip install llama-cpp-python --no-cache-dir'
+    
+    # Run the command using subprocess
+    try:
+        subprocess.run(command, shell=True, check=True)
+        print("Command executed successfully.")
+    except subprocess.CalledProcessError as e:
+        print(f"Error: {e}")
+        
     st.session_state.retriever = init_retriever()
 
 # Function to apply rounded edges using CSS