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 |
|