Spaces:
Sleeping
Sleeping
Update classification_chain.py
Browse files- 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
|