Update app.py

app.py

@@ -23,7 +23,7 @@ os.environ["TOKENIZERS_PARALLELISM"] = "false"
 
 # Predefined values
 predefined_pdf = "t6.pdf"
-predefined_llm = "tiiuae/falcon-7b-instruct"  # Use a smaller model for faster responses
+predefined_llm = "TinyLlama/TinyLlama-1.1B-Chat-v1.0"  # Use a smaller model for faster responses
 
 def load_doc(list_file_path, chunk_size, chunk_overlap):
     loaders = [PyPDFLoader(x) for x in list_file_path]
@@ -156,7 +156,7 @@ collection_name = create_collection_name(pdf_filepath)
 vector_db = create_db(doc_splits, collection_name)
 
 # Initialize the LLM chain with threading
-qa_chain = initialize_llmchain(predefined_llm, temperature=0.5, max_tokens=512, top_k=5, vector_db=vector_db)
+qa_chain = initialize_llmchain(predefined_llm, temperature=0.6, max_tokens=512, top_k=5, vector_db=vector_db)
 
 # Check if qa_chain is properly initialized
 if qa_chain is None: