pratikshahp commited on
Commit
0986e5b
·
verified ·
1 Parent(s): 66632c6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -30,7 +30,7 @@ prohibited_topics = [
30
  def chatbot_response_with_guardrails(message):
31
  try:
32
  # Step 1: Generate raw response using Mistral model
33
- raw_response = llm(message)
34
 
35
  # Step 2: Use TogetherAI's guardrail model to check the response
36
  response = client.completions.create(
 
30
  def chatbot_response_with_guardrails(message):
31
  try:
32
  # Step 1: Generate raw response using Mistral model
33
+ raw_response = llm.invoke(message)
34
 
35
  # Step 2: Use TogetherAI's guardrail model to check the response
36
  response = client.completions.create(