Amey9766 commited on
Commit
4e90c33
·
verified ·
1 Parent(s): 9c7f0b4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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
- "Classify the maintenance request into one of the following categories: "
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
  )