Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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-
|
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 |
|