Update app.py
Browse files
app.py
CHANGED
|
@@ -18,8 +18,8 @@ model.to("cpu")
|
|
| 18 |
def classify(text):
|
| 19 |
# Strong classification prompt to force label output
|
| 20 |
prompt = (
|
| 21 |
-
"
|
| 22 |
-
"urgent, routine, cosmetic.\n\n"
|
| 23 |
f"Request: {text}\n"
|
| 24 |
"Category:"
|
| 25 |
)
|
|
|
|
| 18 |
def classify(text):
|
| 19 |
# Strong classification prompt to force label output
|
| 20 |
prompt = (
|
| 21 |
+
"You are a maintenance request classifier. "
|
| 22 |
+
"Your job is to output ONLY ONE WORD: urgent, routine, or cosmetic.\n\n"
|
| 23 |
f"Request: {text}\n"
|
| 24 |
"Category:"
|
| 25 |
)
|