Phoenix21 commited on
Commit
acd8321
·
verified ·
1 Parent(s): 14e74d1

Update pipeline.py

Browse files
Files changed (1) hide show
  1. pipeline.py +1 -1
pipeline.py CHANGED
@@ -395,7 +395,7 @@ def run_pipeline(query: str) -> str:
395
  return "Sorry, this query contains harmful or inappropriate content."
396
 
397
  # Classify
398
- classification = classify_query(moderation_result.original_text)
399
 
400
  # If out-of-scope, refuse
401
  if classification == "OutOfScope":
 
395
  return "Sorry, this query contains harmful or inappropriate content."
396
 
397
  # Classify
398
+ classification = classificatin_chain.invoke({"query":"moderation_result.original_text"})
399
 
400
  # If out-of-scope, refuse
401
  if classification == "OutOfScope":