added postprocessing to return multiple answers

app.py

@@ -29,6 +29,21 @@ def choose_model():
             return model_cp, model_cp
 
 
+def postprocess(resp):
+  """Function filters postprocessed model output to remove answers that
+  are substrings of another answer."""
+  answers = [answer['answer'] for answer in resp]
+
+  answers.sort(key=lambda s: len(s), reverse=True)
+  filtered_answers = []
+  for s in answers:
+    if not any([s in o for o in filtered_answers]):
+      filtered_answers.append(s)
+
+  filtered_resp = list(filter(lambda r: r['answer'] in filtered_answers, resp))
+  return filtered_resp
+
+
 # Page config
 title = "Recipe Improver"
 icon = "🍣"
@@ -47,14 +62,17 @@ with st.expander("Upload csv file"):
     uploaded_file = st.file_uploader("Choose a csv file", type="csv", key='file_uploader')
 
 
-# If file is uploaded, run inference
+# If file is uploaded, run inference - QA pipelines can't run batch ATM
 if uploaded_file is not None:
     df = load_data(uploaded_file)
 
     # Run inference on first example
     first_example = df['review'][0]
     question = "How was the recipe improved?"
-    answer = question_answer(question=question, context=first_example)
+    resp = question_answer(question=question,
+                             context=first_example,
+                             top_k=5)
+    filtered_resp = postprocess(resp)
 
     # Present results
     st.markdown(f"""
@@ -64,8 +82,16 @@ if uploaded_file is not None:
 
     {first_example}
 
-    The answer given to the question 'how to improve this recipe?' was:
+    The question was:
+
+    {question}
+
+    The unfiltered answers were:
+
+    {resp}
+
+    The filtered answers were:
 
-    {answer['answer']}
+    {filtered_resp}
 
     """)