Upload app.py

app.py

@@ -131,7 +131,9 @@ def gpt3_qa(query, answer):
 
 st.title("Abstractive Question Answering")
 
-st.write("The app uses the quarterly earnings call transcripts for 10 companies (Apple, AMD, Amazon, Cisco, Google, Microsoft, Nvidia, ASML, Intel, Micron) for the years 2016 to 2020.")
+st.write(
+    "The app uses the quarterly earnings call transcripts for 10 companies (Apple, AMD, Amazon, Cisco, Google, Microsoft, Nvidia, ASML, Intel, Micron) for the years 2016 to 2020."
+)
 
 query_text = st.text_input("Input Query", value="Who is the CEO of Apple?")
 
@@ -147,25 +149,21 @@ encoder_model = st.selectbox("Select Encoder Model", encoder_models_choice)
 
 # Choose decoder model
 
-decoder_models_choice = ["GPT3 (QA_davinci)", "GPT3 (summary_davinci)", "T5", "FLAN-T5"]
+decoder_models_choice = ["FLAN-T5", "T5", "GPT3 (QA_davinci)", "GPT3 (summary_davinci)"]
 
 decoder_model = st.selectbox("Select Decoder Model", decoder_models_choice)
 
 
 if encoder_model == "MPNET":
     # Connect to pinecone environment
-    pinecone.init(
-        api_key=st.secrets["pinecone_mpnet"], environment="us-east1-gcp"
-    )
+    pinecone.init(api_key=st.secrets["pinecone_mpnet"], environment="us-east1-gcp")
     pinecone_index_name = "week2-all-mpnet-base"
     pinecone_index = pinecone.Index(pinecone_index_name)
     retriever_model = get_mpnet_embedding_model()
 
 elif encoder_model == "SGPT":
     # Connect to pinecone environment
-    pinecone.init(
-        api_key=st.secrets["pinecone_sgpt"], environment="us-east1-gcp"
-    )
+    pinecone.init(api_key=st.secrets["pinecone_sgpt"], environment="us-east1-gcp")
     pinecone_index_name = "week2-sgpt-125m"
     pinecone_index = pinecone.Index(pinecone_index_name)
     retriever_model = get_sgpt_embedding_model()