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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -16,8 +16,9 @@ llm = HuggingFaceEndpoint(
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):
 
16
  max_new_tokens=100
17
  )
18
 
19
+ # Use a pipeline as a high-level helper
20
+
21
+ content_filter = pipeline("text-generation", model="meta-llama/LlamaGuard-7b")
22
 
23
  # Function to handle chatbot response and guardrails
24
  def chatbot_response_with_guardrails(message):