Refactor classification prompt to limit categories to Transaction and OTP

app.py

@@ -137,8 +137,7 @@ def write_message(data: dict, header: str = Header()):
     prompt = """You are a smart assistant adept at classifying different messages. \
     You will be penalized heavily for incorrect classification. \
     Your task is to classify the message into one of the following categories: \
-    Transaction, OTP, Promotional, Scheduled, Non-Transaction. \
-    Consider Salary Credit, Interest Credit, Redemptions as non-transactional messages.
+    Transaction or OTP. \
     Output the classification in a structured format like below. \
     {"classification": "OTP"} \
     """