Phoenix21 commited on
Commit
14e74d1
·
verified ·
1 Parent(s): c555c22

Update classification_chain.py

Browse files
Files changed (1) hide show
  1. classification_chain.py +0 -2
classification_chain.py CHANGED
@@ -16,8 +16,6 @@ def get_classification_chain() -> LLMChain:
16
  model="Gemma2-9b-It",
17
  groq_api_key=os.environ["GROQ_API_KEY"] # must be set in environment
18
  )
19
- prompt=
20
-
21
  # Build an LLMChain
22
  classification_chain = classification_prompt|chat_groq_model|output_parser
23
  return classification_chain
 
16
  model="Gemma2-9b-It",
17
  groq_api_key=os.environ["GROQ_API_KEY"] # must be set in environment
18
  )
 
 
19
  # Build an LLMChain
20
  classification_chain = classification_prompt|chat_groq_model|output_parser
21
  return classification_chain