Update app.py

app.py

@@ -56,8 +56,6 @@ with st.expander("How to format your CSV", expanded=False):
 - Optional columns: `REAGENT`, `SOLVENT`, `CATALYST`.  
 - If a field lists multiple compounds, separate them with a dot (`.`).  
 - For details, download **demo_reaction_data.csv** and check its contents.  
-- Output contains predicted product **SMILES** and the sum of log-likelihoods for each prediction,  
-  sorted by log-likelihood (index 0 is most probable).
 """
         )
     elif task == "retrosynthesis prediction":
@@ -67,8 +65,6 @@ with st.expander("How to format your CSV", expanded=False):
 - No optional columns are used.  
 - If a field lists multiple compounds, separate them with a dot (`.`).  
 - For details, download **demo_retro_data.csv** and check its contents.  
-- Output contains predicted **sets of reactant SMILES** and their log-likelihoods,  
-  sorted by log-likelihood (index 0 is most probable).
 """
         )
     else:  # yield prediction
@@ -260,14 +256,29 @@ with left:
 
 with right:
     st.subheader("Notes")
-    st.markdown(
-        f"""
-- Approximate time: about **15 seconds per reaction** when `beam size = 5` (varies by hardware).  
-- Results include the **sum of log-likelihoods** per prediction and are **sorted** by that value.
-"""
-    )
+    if task == "product prediction":
+        st.markdown(
+            f"""
+    - Approximate time: about **3 seconds per reaction** when `beam size = 5` (varies by hardware).  
+    - Output contains predicted **sets of reactant SMILES** and their log-likelihoods, sorted by log-likelihood (index 0 is most probable).
+    """
+        )
+    elif task == "retrosynthesis prediction":
+        st.markdown(
+            f"""
+    - Approximate time: about **5 seconds per reaction** when `beam size = 5` (varies by hardware).  
+    - Output contains predicted **sets of reactant SMILES** and their log-likelihoods, sorted by log-likelihood (index 0 is most probable).
+    """
+        )
+    else:  # yield prediction
+        st.markdown(
+            f"""
+    - Approximate time: about **0.25 seconds per reaction** when `batch size = 1` (varies by hardware).  
+    - Output contains predicted **reaction yield** on a **0–100% scale**.
+    """
+        )
     st.info(
-        "If you encounter CUDA OOM issues, reduce max lengths or beam size, or switch to CPU."
+        "In this space, CPU is used for inference. So the speed is slower than using a GPU."
     )
 
 # ------------------------------