pratikshahp commited on
Commit
11e8c1c
·
verified ·
1 Parent(s): bb2efc2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -16,8 +16,8 @@ llm = HuggingFaceEndpoint(
16
  max_new_tokens=100
17
  )
18
 
19
- # Initialize content moderation model (e.g., JinaAI ContentFilter or similar)
20
- content_filter = pipeline("text-classification", model="JinaAI/ContentFilter", tokenizer="JinaAI/ContentFilter")
21
 
22
  # Function to handle chatbot response and guardrails
23
  def chatbot_response_with_guardrails(message):
 
16
  max_new_tokens=100
17
  )
18
 
19
+ # Initialize content moderation model (unitary/matti)
20
+ content_filter = pipeline("text-classification", model="unitary/matti")
21
 
22
  # Function to handle chatbot response and guardrails
23
  def chatbot_response_with_guardrails(message):