Update app.py

app.py

@@ -38,7 +38,7 @@ DEVICE = "cuda:0" if torch.cuda.is_available() else "cpu"
 
 @st.cache_data
 def load_data():
-    loader = PyPDFDirectoryLoader("/home/user/app/pdfs/")
+    loader = PyPDFDirectoryLoader("/home/user/app/ML/")
     docs = loader.load()
     return docs
 
@@ -64,22 +64,27 @@ def load_model(_docs):
         quantize_config=None,
     )
 
+    # DEFAULT_SYSTEM_PROMPT = """
+    # You are a helpful, respectful and honest assistant. Always answer as helpfully as possible, while being safe. 
+    # Your answers should not include any harmful, unethical, racist, sexist, toxic, dangerous, or illegal content. 
+    # Please ensure that your responses are socially unbiased and positive in nature. 
+    # Always provide the citation for the answer from the text. 
+    # Try to include any section or subsection present in the text responsible for the answer. 
+    # Provide reference. Provide page number, section, sub section etc.
+    # If a question does not make any sense, or is not factually coherent, explain why instead of answering something not correct. If you don't know the answer to a question, please don't share false information. 
+    # Given a government document that outlines rules and regulations for a specific industry or sector, use your language model to answer questions about the rules and their applicability over time. 
+    # The document may include provisions that take effect at different times, such as immediately upon publication, after a grace period, or on a specific date in the future. 
+    # Your task is to identify the relevant rules and determine when they go into effect, taking into account any dependencies or exceptions that may apply.
+    # The current date is 14 September, 2023. Try to extract information which is closer to this date.
+    # Take a deep breath and work on this problem step-by-step. 
+    # """.strip()
+
+
     DEFAULT_SYSTEM_PROMPT = """
-    You are a helpful, respectful and honest assistant. Always answer as helpfully as possible, while being safe. 
-    Your answers should not include any harmful, unethical, racist, sexist, toxic, dangerous, or illegal content. 
-    Please ensure that your responses are socially unbiased and positive in nature. 
-    Always provide the citation for the answer from the text. 
-    Try to include any section or subsection present in the text responsible for the answer. 
-    Provide reference. Provide page number, section, sub section etc.
-    If a question does not make any sense, or is not factually coherent, explain why instead of answering something not correct. If you don't know the answer to a question, please don't share false information. 
-    Given a government document that outlines rules and regulations for a specific industry or sector, use your language model to answer questions about the rules and their applicability over time. 
-    The document may include provisions that take effect at different times, such as immediately upon publication, after a grace period, or on a specific date in the future. 
-    Your task is to identify the relevant rules and determine when they go into effect, taking into account any dependencies or exceptions that may apply.
-    The current date is 14 September, 2023. Try to extract information which is closer to this date.
-    Take a deep breath and work on this problem step-by-step. 
+    You are a helpful, respectful and honest assistant with knowledge of machine learning, data science, computer science, Python programming language, mathematics, probability and statistics.
+    Take a deep breath and work on the given problem step-by-step. 
     """.strip()
 
-
     def generate_prompt(prompt: str, system_prompt: str = DEFAULT_SYSTEM_PROMPT) -> str:
         return f"""[INST] <<SYS>>{system_prompt}<</SYS>>{prompt} [/INST]""".strip()