implementing app

app.py

@@ -23,6 +23,7 @@ from pytorch_lightning.callbacks import LearningRateMonitor
 from rdkit import Chem
 from rdkit.Chem import AllChem
 from rdkit import DataStructs
+from PIL import Image
 
 model_cls = RCNN
 experiment_path_atoms="./models/atoms_model/"
@@ -32,5 +33,16 @@ dir_list.sort(key=os.path.getctime, reverse=True)
 checkpoint_file_atoms = [f for f in dir_list if "ckpt" in f][0]
 model_atom = model_cls.load_from_checkpoint(checkpoint_file_atoms)
 model_atom.model.roi_heads.score_thresh = 0.65 
+
+
+st.title("Atom Level Entity Detector")
+
+file_name = st.file_uploader("Upload a chemical structure candidate image")
+
+if file_name is not None:
+   col1, col2 = st.columns(2)
+
+   image = Image.open(file_name)
+   col1.image(image, use_column_width=True)
 x = st.slider('Select a value')
 st.write(x, 'squared is', x * x)