Update app.py

app.py

@@ -31,8 +31,11 @@ from mysql.connector import errorcode, pooling
 from dotenv import load_dotenv
 from huggingface_hub import login
 
-# Initialize a zero-shot classification pipeline
-classifier = pipeline("zero-shot-classification", model="facebook/bart-large-mnli")
+# Load model directly
+from transformers import AutoTokenizer, AutoModelForMaskedLM
+
+tokenizer = AutoTokenizer.from_pretrained("sentence-transformers/all-mpnet-base-v2")
+model = AutoModelForMaskedLM.from_pretrained("sentence-transformers/all-mpnet-base-v2")
 
 def parse_command(message):
     candidate_labels = ["filter", "sort", "export", "log"]