Update app.py

app.py

@@ -16,8 +16,9 @@ st.title("Nutrition Chatbot")
 user_input = st.text_input("Ask me about nutrition:")
 
 if user_input:
-    # Encode the input with attention mask
-    inputs = tokenizer.encode(user_input, return_tensors="pt", return_attention_mask=True)
+    # Use encode_plus to get both input_ids and attention_mask
+    inputs = tokenizer.encode_plus(user_input, return_tensors="pt", padding=True, truncation=True)
+    
     input_ids = inputs['input_ids']
     attention_mask = inputs['attention_mask']