Update app.py

app.py

@@ -11,7 +11,7 @@ from transformers import AutoTokenizer, AutoModelForSeq2SeqLM, pipeline
 logging.basicConfig(level=logging.DEBUG)
 
 def load_vector_store():
-    # Directory to load the vector data
+    # Ensure the directory exists
     persist_directory = "./chroma_db"
     if not os.path.exists(persist_directory):
         logging.error(f"The directory '{persist_directory}' does not exist. Please run the ingestion script.")
@@ -30,10 +30,10 @@ def load_llm():
         'text2text-generation',
         model=model,
         tokenizer=tokenizer,
-        max_length=256,
+        max_length=1024,  # Increase max_length to allow longer output
         do_sample=True,
-        temperature=0.3,
-        top_p=0.95
+        temperature=0,  # Adjust temperature to control randomness
+        top_p=0.9         # Adjust top_p to control diversity of the output
     )
     return HuggingFacePipeline(pipeline=pipe)
 
@@ -63,7 +63,7 @@ def main():
     with st.expander("About the App"):
         st.markdown(
             """
-            This is a Generative AI powered Question and Answering app that responds to questions about your PDF File.
+            This is a Generative AI-powered Question and Answering app that responds to questions about your PDF File.
             """
         )
     question = st.text_area("Enter your Question")