Update app.py

app.py

@@ -34,7 +34,9 @@ DISPLAY_DIALOG_LINES = 6
 
 SESSION_STARTED = False
 
-MODEL_NAME="deepset/roberta-base-squad2"
+# MODEL_NAME="deepset/roberta-base-squad2"
+
+MODEL_NAME="BEE-spoke-data/TinyLlama-3T-1.1bee"
 
 ##################################################################################
 def extract_pdf_text(pdf_docs):
@@ -107,7 +109,7 @@ def prepare_conversation(vectorstore):
 
 ##################################################################################
 def process_user_question(user_question):
-    st.write( ('process_user_question called: '+user_question), unsafe_allow_html=True)
+
 
     # if not SESSION_STARTED:
     #     print('No Session')
@@ -183,7 +185,7 @@ def process_user_question(user_question):
     #
     #         results_string += ("<p>" + "-- " + message.content + "</p>")
 
-    st.write('start pipelene', unsafe_allow_html=True)
+    st.write(('Question: ' + user_question), unsafe_allow_html=True)
 
     from transformers import pipeline
 
@@ -198,9 +200,10 @@ def process_user_question(user_question):
     # context = "France is a country located in Western Europe. It is bordered by the Atlantic Ocean to the west, the Mediterranean Sea to the south, and Belgium, Luxembourg, Germany, Switzerland, Italy, and Spain to the east and north."
 
     context = "You are an expert Apiarist and answer all questions regarding Honeybees."
+    # context = " "
 
     # Ask the question
-    answer = nlp(question=user_question, context=context, model=model_name)
+    answer = nlp(question= ('Question: '+user_question), context=context, model=model_name)
 
     # Print the answer
     print(f"Answer: {answer['answer']}")