shresthasingh commited on
Commit
755d9c1
·
verified ·
1 Parent(s): 78e9098

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -42,14 +42,14 @@ def classify_message(message: str) -> str:
42
  messages=[
43
  {
44
  "role": "system",
45
- "content": "Classify the following message into one of these categories: create_order, cancel_order, check_order_status, create_invoice, get_invoice_details, create_payment. Respond only with the category name.",
46
  },
47
  {
48
  "role": "user",
49
  "content": message,
50
  },
51
  ],
52
- model="meta-llama/Meta-Llama-3.1-8B-Instruct-Turbo",
53
  )
54
  return classify.choices[0].message.content.strip()
55
 
 
42
  messages=[
43
  {
44
  "role": "system",
45
+ "content": "Strictly only Classify the following message into one of these categories: create_order, cancel_order, check_order_status, create_invoice, get_invoice_details, create_payment. Respond only with the category name and nothing else , even if its ambgious , give your best classification",
46
  },
47
  {
48
  "role": "user",
49
  "content": message,
50
  },
51
  ],
52
+ model="meta-llama/Meta-Llama-3.1-70B-Instruct-Turbo",
53
  )
54
  return classify.choices[0].message.content.strip()
55