Update app.py

app.py

@@ -5,6 +5,7 @@ from langchain.embeddings.openai import OpenAIEmbeddings
 from langchain.text_splitter import CharacterTextSplitter
 from langchain.vectorstores import FAISS 
 from transformers import TFAutoModelForQuestionAnswering, AutoTokenizer, pipeline
+from PIL import Image
 
 os.environ["OPENAI_API_KEY"] = "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"
 
@@ -73,12 +74,12 @@ with st.sidebar:
     selected = option_menu("Model selection", ["Roberta base squad2", "Bert finetuned squad"], 
         icons=['box-fill', 'box-fill'], menu_icon="cast", default_index=0)
 
-
+image = Image.open('Swinburne_Logo.png')
+st.image(image)
 st.markdown('<p class="big-font">QnA for Swinburne\'s Bachelor of Computer Science progrom</p>', unsafe_allow_html=True)
+st.write("- ", selected)
 
 if selected == "Roberta base squad2":
-    
-    st.write("- ", selected)
     text0 = st.text_area("Type question (Eg. What is the duration of the Bachelor of Computer Science program?):")
     if text0:
         #######
@@ -91,8 +92,6 @@ if selected == "Roberta base squad2":
 
 
 elif selected == "Bert finetuned squad":
-    st.markdown('<p class="big-font">QnA for Swinburne\'s Bachelor of Computer Science progrom</p>', unsafe_allow_html=True)
-    st.write("- ", selected)
     text1 = st.text_area("Type question (Eg. What is the duration of the Bachelor of Computer Science program?): ")
     if text1:
         # Fed in the question to the model