Update classification_chain.py

classification_chain.py

@@ -16,6 +16,7 @@ def get_classification_chain() -> LLMChain:
         model="Gemma2-9b-It",
         groq_api_key=os.environ["GROQ_API_KEY"]  # must be set in environment
     )
+    output_parser=StrOutputParser()
     # Build an LLMChain
     classification_chain = classification_prompt|chat_groq_model|output_parser
     return classification_chain