Update main.py

main.py

@@ -312,7 +312,25 @@ async def fast_classify_text(statement: str = Form(...)):
         # Handle general errors
         return JSONResponse(content=f"Error in classification pipeline: {str(e)}", status_code=500)
 
-app.post("/fast_classify_v2/", description="Quickly classify text into predefined categories with confidence scores.")
+# Predefined classifications
+labels = [
+    "All Pricing copy quote requested",
+    "Change to quote",
+    "Change to quote & Status Check",
+    "Change to quote (Items missed?)",
+    "Confirmation",
+    "Copy quote requested",
+    "Cost copy quote requested",
+    "MRSP copy quote requested",
+    "MSRP & All Pricing copy quote requested",
+    "MSRP & Cost copy quote requested",
+    "No narrative in email",
+    "Notes not clear",
+    "Retail copy quote requested",
+    "Status Check (possibly)"
+]
+
+@app.post("/fast_classify_v2/", description="Quickly classify text into predefined categories.")
 async def fast_classify_text(statement: str = Form(...)):
     try:
         # Use run_in_executor to handle the synchronous model call asynchronously