Update app.py

app.py

@@ -218,6 +218,7 @@ with tab1:
     
         # Keep only necessary columns
         df = df[['content','Class']]
+        df1=df[['content']]
         
         #show Classification Results
         st.subheader("Classification Results")
@@ -255,7 +256,7 @@ with tab2:
     st.markdown('<div class="card">', unsafe_allow_html=True)
     st.header("Ask Questions Based on Uploaded News Content File")
     st.write("Ask questions about news content and get answers from our AI model.")
-
+     
     #check file is uploaded
     if uploaded_file is not None:
         context = ' '.join(df['content'].tolist())
@@ -284,7 +285,7 @@ with tab2:
     #generate the answer based on selected news content using the given model
     st.markdown("---")
     st.header("Ask Questions Based on Your News Content")
-    context_1 = st.selectbox("Choose an article for the question:", df['content'].tolist())
+    context_1 = st.selectbox("Choose an article for the question:", df1['content'].tolist())
     
     question_1 = st.text_input("Enter your question:", key="question_input")
     if st.button("Get Answer", key="get_answer_1"):