filter results

app.py

@@ -8,8 +8,17 @@ def answer_question(subreddit, num_posts, category, question):
     db = create_vector_db(docs)
     qa_chain = create_qa_chain(db)
     result = qa_chain(question)
-    print("test", result)
-    return result["result"]
+    output = result["result"]
+
+    # Remove instruction sentence
+    output = output.replace(
+        "Use the following pieces of context to answer the question at the end. If you don't know the answer, just say that you don't know, don't try to make up an answer.",
+        "",
+    )
+    # Remove everything after "Example:"
+    output = output.split("Example:")[0]
+    
+    return output.strip()
 
 
 # Gradio UI