center app

app.py

@@ -21,7 +21,7 @@ data_path = os.path.join(base_path, 'data')
 checkpoint_path = os.path.join(base_path, 'checkpoints/lpo/cv2_test_fold6_1402/model_updated.t7')
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
-st.set_page_config(layout="wide")
+st.set_page_config(layout="centered")
 
 st.title('HyperDTI: Robust Task-Conditioned Modeling of Drug-Target Interactions\n')
 st.markdown('')
@@ -51,7 +51,7 @@ def about_page():
         """
     )
 
-    st.image('figures/hyper-dti.png', caption='Overview of HyperPCM architecture.')
+    st.image('figures/hyper-dti.gif', caption='Overview of HyperPCM architecture.')
     
 
 def retrieval():
@@ -78,7 +78,7 @@ def retrieval():
     
     with col2:
         selected_encoder = st.selectbox(
-                'Select target encoder',('SeqVec')
+                'Select target encoder',(['SeqVec'])
             )
         if sequence:
             st.image('figures/protein_encoder_done.png')
@@ -114,7 +114,7 @@ def retrieval():
     
     with col4:
         selected_encoder = st.selectbox(
-                'Select drug encoder',('CDDD')
+                'Select drug encoder',(['CDDD'])
             )
         st.image('figures/molecule_encoder_done.png')
         st.success('Encoding complete.')