Update app.py

app.py

@@ -22,11 +22,22 @@ app = Flask(__name__)
 CORS(app, resources={r"/send_message": {"origins": ["http://localhost:3000", "https://main.dbn2ikif9ou3g.amplifyapp.com"]}})
   
 # Load zero-shot classification pipeline
-classifier = pipeline("zero-shot-classification")
+#classifier = pipeline("zero-shot-classification")
 
  # Load Sentence-BERT model
 bertmodel = SentenceTransformer('all-MiniLM-L6-v2')  # Lightweight, efficient model; choose larger if needed
-        
+
+# Load model with accelerator
+classifier = pipeline(
+    "zero-shot-classification",
+    model="facebook/bart-large-mnli",
+    revision="d7645e1",
+    device=accelerator.device  # Ensures correct device placement
+)
+
+# Move model to correct device
+classifier.model = accelerator.prepare(classifier.model)
+
 # Global variables for model and tokenizer
 model = None
 tokenizer = None