log code change directory 9

code/qa.py

@@ -3,6 +3,11 @@ from transformers import pipeline
 import pandas as pd
 import datetime
 
+@st.cache(allow_output_mutation = True)
+def load_model():
+    question_answerer = pipeline('question-answering')
+    return question_answerer
+
 def load_file():
     """Load text from file"""
     uploaded_file = st.file_uploader("Upload Files",type=['txt'])
@@ -12,62 +17,55 @@ def load_file():
             raw_text = str(uploaded_file.read(),"utf-8")
         return raw_text
 
+# Loading Model
+question_answerer =load_model()
 
-if __name__ == "__main__":
-
-    # App title and description
-    st.title("Answering questions from text")
-    st.write("Upload text, pose questions, get answers")
+# App title and description
+st.title("Answering questions from text")
+st.write("Upload text, pose questions, get answers")
 
-    # Load file
-    raw_text = load_file()
-    if raw_text != None and raw_text != '':
+# Load file
+raw_text = load_file()
+if raw_text != None and raw_text != '':
 
-        # Display text
-        with st.expander("See text"):
-            st.write(raw_text)
+    # Display text
+    with st.expander("See text"):
+        st.write(raw_text)
 
-        # Perform question answering
-        question_answerer = pipeline('question-answering')
+    # Perform question answering
+    # question_answerer = pipeline('question-answering')
 
-        answer = ''
-        question = st.text_input('Ask a question')
+    answer = ''
+    question = st.text_input('Ask a question')
 
-        if question != '' and raw_text != '':
-            answer = question_answerer({
-                'question': question,
-                'context': raw_text
-            })
+    if question != '' and raw_text != '':
+        answer = question_answerer({
+            'question': question,
+            'context': raw_text
+        })
 
-            st.write("Answer: {}".format(answer["answer"]))
-            st.write("Confidence score: {}".format(round(answer["score"],2)))
-            import os
-            # st.write(os.getcwd())
-            # st.write(os.listdir('.'))
-            # st.write(os.listdir('../'))            
-            # log_file_object = open('../logs/log_file.tsv','a')
-            # log_file_object.write(str(datetime.datetime.now())+'\t'+str(raw_text)+'\t'+str(answer["answer"])+'\t'+str(answer["score"]))
-            # log_file_object.write('\n')
-            # log_file_object.close()
-            output_path = "results/df_log_file.tsv"
-            try:
-                res_df = pd.DataFrame({"TimeStamp":[str(datetime.datetime.now())],\
-                    "Input":[str(raw_text)],"Answer":[str(answer["answer"])],"Score":[str(answer["score"])]})
-                st.write("PASS1")
-            except:
-                st.write("FAIL1")
-            try:
-                res_df.to_csv(output_path, mode='a', index=False, sep="\t", header= not os.path.exists(output_path))
-                st.write("PASS2")
-            except:
-                st.write("FAIL2")
-            import os
-            #st.write(os.listdir("logs"))
-            for i in os.listdir("."):
-                try:
-                    st.write(i,os.listdir(i))
-                    st.write("="*50)
-                except:
-                    pass
-            st.write(pd.read_csv("logs/df_log_file.tsv",sep="\t").head(10))
+        st.write("Answer: {}".format(answer["answer"]))
+        st.write("Confidence score: {}".format(round(answer["score"],2)))
+        import os
+        # st.write(os.getcwd())
+        # st.write(os.listdir('.'))
+        # st.write(os.listdir('../'))            
+        # log_file_object = open('../logs/log_file.tsv','a')
+        # log_file_object.write(str(datetime.datetime.now())+'\t'+str(raw_text)+'\t'+str(answer["answer"])+'\t'+str(answer["score"]))
+        # log_file_object.write('\n')
+        # log_file_object.close()
+        output_path = "results/df_log_file.tsv"
+        try:
+            res_df = pd.DataFrame({"TimeStamp":[str(datetime.datetime.now())],\
+                "Input":[str(raw_text)],"Answer":[str(answer["answer"])],"Score":[str(answer["score"])]})
+            st.write("PASS1")
+        except:
+            st.write("FAIL1")
+        try:
+            res_df.to_csv(output_path, mode='a', index=False, sep="\t", header= not os.path.exists(output_path))
+            st.write("PASS2")
+        except:
+            st.write("FAIL2")
+        import os
+        st.dataframe(pd.read_csv("results/df_log_file.tsv",sep="\t").head(10))