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
|