Update app.py

app.py

@@ -89,6 +89,7 @@ decoder_models_choice = [
     "GPT3 - (text-davinci-003)",
     "T5",
     "FLAN-T5",
+    "GPT-J"
 ]
 
 with st.sidebar:
@@ -181,6 +182,17 @@ elif decoder_model == "FLAN-T5":
         st.subheader("Answer:")
         st.write(flan_t5_pipeline(generated_text)[0]["summary_text"])
 
+elif decoder_model == "GPTJ":
+    flan_t5_pipeline = get_gptj_model()
+    output_text = []
+    for context_text in context_list:
+        output_text.append(gptj_pipline(context_text)[0]["summary_text"])
+    generated_text = ". ".join(output_text)
+    with col2:
+        st.subheader("Answer:")
+        st.write(flan_t5_pipeline(generated_text)[0]["summary_text"])
+
+
 with col1:
     with st.expander("See Retrieved Text"):
         for context_text in context_list: