Update app.py

app.py

@@ -94,6 +94,8 @@ def load_model():
         quantize_config=None,
     )
 
+    print('model done')
+    
     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 from which answer is taken.
 
@@ -111,6 +113,8 @@ def load_model():
 
     llm = HuggingFacePipeline(pipeline=text_pipeline, model_kwargs={"temperature": 0.2})
 
+    print('llm done')
+    
     SYSTEM_PROMPT = "Use the following pieces of context to answer the question at the end. If you don't know the answer, just say that you don't know, don't try to make up an answer."
 
     template = generate_prompt("""{context}  Question: {question} """,system_prompt=SYSTEM_PROMPT,) #Enter memory here!
@@ -129,6 +133,8 @@ def load_model():
                               #input_key="question",
                               #return_messages=True)
                               },)
+
+    print('load done')
     return qa_chain