Update app.py

app.py

@@ -37,11 +37,20 @@ predictions = list(xgb.predict(data))
 
 output = pd.Series(data=predictions, index=data.index, name="XGB_Score")
 df_total = pd.concat([data, output], axis=1)
-
+df_total['XGB_Score'] = round(df_total['XGB_Score'], 2)
 
 st.title('Blood Pressure Gene Prioritisation Post-GWAS')
 st.markdown("""
 A machine learning pipeline for predicting disease-causing genes post-genome-wide association study in blood pressure.
 """)
 
-AgGrid(df_total)
+gene_input = st.text_input('Input HGNC Gene') 
+df = df_total[df_total.index == gene_input] 
+st.dataframe(df)
+
+
+st.markdown("""
+Total Gene Prioritisation Results:
+""")
+
+st.dataframe(df_total)