Update app.py

app.py

@@ -18,13 +18,20 @@ user_input = st.text_input("Ask me about nutrition:")
 if user_input:
     # 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']
     
-    # Generate output with attention mask and pad token ID
-    outputs = model.generate(input_ids, attention_mask=attention_mask, max_length=50)
-    answer = tokenizer.decode(outputs[0], skip_special_tokens=True)
+    # Print the input tensors for debugging
+    st.write("Input IDs:", input_ids)
+    st.write("Attention Mask:", attention_mask)
     
-    # Display answer
-    st.write("Answer:", answer)
+    # Generate output with attention mask and pad token ID
+    try:
+        outputs = model.generate(input_ids, attention_mask=attention_mask, max_length=50)
+        st.write("Model Output (Raw):", outputs)  # Debugging model raw output
+        
+        # Decode the output and display
+        answer = tokenizer.decode(outputs[0], skip_special_tokens=True)
+        st.write("Answer:", answer)
+    except Exception as e:
+        st.write("Error generating output:", str(e))