Spaces:
Sleeping
Sleeping
Update pipeline.py
Browse files- 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 =
|
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":
|