fixed numpy error added time calc for env setup

ProtHGT_app.py

@@ -1,18 +1,20 @@
 import os
 import streamlit as st
+import time
+import streamlit.components.v1 as components
+import pandas as pd
+
 
 with st.spinner("Please wait while we prepare the environment. This may take a few minutes only on the first run..."):
     # Run setup script if not already executed
     if not os.path.exists(".setup_done"):
+        start_time = time.time()
         os.system("bash setup.sh")
+        end_time = time.time()
+        st.success(f"Environment prepared in {end_time - start_time:.2f} seconds")
         with open(".setup_done", "w") as f:
             f.write("done")
 
-import streamlit.components.v1 as components
-import os
-import time
-import pandas as pd
-
 from run_prothgt_app import *
 
 def convert_df(df):

run_prothgt_app.py

@@ -103,7 +103,7 @@ def _create_prediction_df(predictions, heterodata, protein_ids, go_category):
             'Protein': protein_id,
             'GO_category': go_category_dict[go_category],
             'GO_term': list(heterodata[go_category]['id_mapping'].keys()),
-            'Probability': protein_predictions.numpy()
+            'Probability': protein_predictions.tolist()
         })
         all_predictions.append(prediction_df)