Update app.py

app.py

@@ -8,10 +8,10 @@ from chat import generate_response
 
 if "tokenizer" not in st.session_state:
     st.session_state["tokenizer"] = AutoTokenizer.from_pretrained(
-        "./generative_model/LaMini-Flan-T5-783M"
+        "MBZUAI/LaMini-Flan-T5-248M"
     )
     st.session_state["model"] = AutoModelForSeq2SeqLM.from_pretrained(
-        "./generative_model/LaMini-Flan-T5-783M"
+        "MBZUAI/LaMini-Flan-T5-248M"
     )
 
 st.title("B-Bot : Bibek's Personal Chatbot")
@@ -35,12 +35,13 @@ if user_input:
     tokenizer = st.session_state["tokenizer"]
     model = st.session_state["model"]
     output = generate_response(user_input)
-    prompt_template = "\nPlease make meaningful sentence and try to be descriptive as possible, ending with proper punctuations. If you don't have descriptive answers from the available prompt, write sorry and advise them to contact Bibek directly."  # NoQA
+    prompt_template = "\nBased on the above content, try to answer the following question.\n\n"
+    end_prompt = "Please make meaningful sentence and try to be descriptive as possible, ending with proper punctuations. If you think, there is good descriptive answers to the question from the above content, write sorry and advise them to contact Bibek directly.\n"  # NoQA"
     short_response_template = "\nIf your response is very short like 1 or 2 sentence, add a followup sentence like 'Let me know if there's anything else I can help you with. or If there's anything else I can assist with, please don't hesitate to ask. I mean something similar in polite way."  # NoQA
-
+    input = output + prompt_template + user_input + end_prompt
     start = time.time()
     input_ids = tokenizer(
-        output + user_input + prompt_template + short_response_template,
+        input,
         return_tensors="pt",
     ).input_ids