Update app.py

app.py

@@ -9,10 +9,10 @@ def load_data():
 
 data = load_data()
 
-# Load NLP model for intent detection
+# Load NLP model for intent detection using DeepSeek
 @st.cache_resource
 def load_nlp_model():
-    return pipeline("text-classification", model="facebook/bart-large-mnli")
+    return pipeline("zero-shot-classification", model="deepseek-ai/deepseek-llm-7b-chat")
 
 classifier = load_nlp_model()
 
@@ -24,7 +24,13 @@ if user_input:
     # Detect intent
     labels = ["coverage explanation", "plan recommendation"]
     result = classifier(user_input, candidate_labels=labels)
-    intent = result["labels"][0]  # Get the most likely intent
+    
+    # Check if DeepSeek returns results properly
+    if "labels" in result:
+        intent = result["labels"][0]  # Get the most likely intent
+    else:
+        st.write("Intent detection failed. Please try again.")
+        intent = None
 
     if intent == "coverage explanation":
         st.subheader("Coverage Details")