Spaces:
Sleeping
Sleeping
Update pipeline.py
Browse files- pipeline.py +4 -2
pipeline.py
CHANGED
@@ -166,14 +166,16 @@ def classify_query(query: str) -> str:
|
|
166 |
query_input = QueryInput(query=query)
|
167 |
|
168 |
# Quick pattern-based approach for 'Wellness'
|
169 |
-
wellness_keywords = ["box breathing", "meditation", "yoga", "mindfulness", "breathing exercises"]
|
|
|
170 |
if any(keyword in query_input.query.lower() for keyword in wellness_keywords):
|
171 |
return "Wellness"
|
172 |
|
173 |
# Use chain for everything else
|
174 |
class_result = classification_chain.invoke({"query": query_input.query})
|
175 |
print(class_result)
|
176 |
-
classification = class_result.get("text", "").strip()
|
|
|
177 |
|
178 |
return classification if classification != "" else "OutOfScope"
|
179 |
except ValidationError as e:
|
|
|
166 |
query_input = QueryInput(query=query)
|
167 |
|
168 |
# Quick pattern-based approach for 'Wellness'
|
169 |
+
# wellness_keywords = ["box breathing", "meditation", "yoga", "mindfulness", "breathing exercises"]
|
170 |
+
wellness_keywords=[]
|
171 |
if any(keyword in query_input.query.lower() for keyword in wellness_keywords):
|
172 |
return "Wellness"
|
173 |
|
174 |
# Use chain for everything else
|
175 |
class_result = classification_chain.invoke({"query": query_input.query})
|
176 |
print(class_result)
|
177 |
+
# classification = class_result.get("text", "").strip()
|
178 |
+
classification=class_result
|
179 |
|
180 |
return classification if classification != "" else "OutOfScope"
|
181 |
except ValidationError as e:
|