Update app.py

app.py

@@ -3,7 +3,7 @@
 # streamlit run app.py
 #
 # You also need to install the following libraries:
-# pip install streamlit torch transformers bitsandbytes rdkit py3Dmol pandas
+# pip install streamlit torch transformers bitsandbytes rdkit-pypi py3Dmol pandas
 
 import streamlit as st
 import streamlit.components.v1 as components
@@ -12,7 +12,6 @@ from transformers import AutoModelForMaskedLM, AutoTokenizer, pipeline, BitsAndB
 from rdkit import Chem
 from rdkit.Chem import Draw, AllChem
 from rdkit.Chem.Draw import MolToImage
-import py3Dmol
 import pandas as pd
 import logging
 
@@ -250,4 +249,5 @@ with tab2:
                 with col2:
                     st.subheader("3D Interactive Structure")
                     html_3d = generate_3d_view_html(smiles_input_viewer)
-                    components.html(html_3d, height=470)
+                    components.html(html_3d, height=470)
+