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:
|