Update app.py

app.py

@@ -65,11 +65,8 @@ if Run_Button == True:
     qna_pipeline = setModel(model_checkpoint)
     output = qna_pipeline(question=selected_question, context=selected_context)
 
-    output["answer"] = str(output["answer"])
 
     df = pd.DataFrame.from_dict(output)
-    cols_to_keep = ['answer','score','start','end']
-    df_final = df[cols_to_keep]
-    
+        
     st.header("Answer")
-    st.dataframe(df_final)
+    st.dataframe(df)