temporarily disable display dti

app.py

@@ -148,40 +148,6 @@ def retrieval():
         selected_encoder = st.selectbox(
             'Select encoder for protein target',('None', 'SeqVec', 'UniRep', 'ESM-1b', 'ProtT5')
         )
-        if selected_encoder == 'SeqVec':
-            from bio_embeddings.embed import SeqVecEmbedder
-            encoder = SeqVecEmbedder()
-            embeddings = encoder.embed_batch([sequence])
-            for emb in embeddings:
-                embedding = encoder.reduce_per_protein(emb)
-                break
-        elif selected_encoder == 'UniRep':
-            from jax_unirep.utils import load_params
-            params = load_params() 
-            from jax_unirep.featurize import get_reps                               
-            embedding, h_final, c_final = get_reps([sequence])
-            embedding = embedding.mean(axis=0)
-        elif selected_encoder == 'ESM-1b':
-            from bio_embeddings.embed import ESM1bEmbedder
-            encoder = ESM1bEmbedder()
-            embeddings = encoder.embed_batch([sequence])
-            for emb in embeddings:
-                embedding = encoder.reduce_per_protein(emb)
-                break
-        elif selected_encoder == 'ProtT5':
-            from bio_embeddings.embed import ProtTransT5XLU50Embedder
-            encoder = ProtTransT5XLU50Embedder()
-            embeddings = encoder.embed_batch([sequence])
-            for emb in embeddings:
-                embedding = encoder.reduce_per_protein(emb)
-                break
-        else: 
-            st.write('No pre-trained version of HyperPCM is available for the chosen encoder.')
-            embedding = None
-        if embedding is not None:
-            st.write(f'{selected_encoder} embedding')
-            st.write(embedding)
-    
     
     st.markdown('### Retrieval of top-5 drug coupound from ChEMBL:')
     col1, col2, col3, col4, col5 = st.columns(5)
@@ -282,7 +248,7 @@ def display_protein():
 
 page_names_to_func = {
     'About': about_page,
-    'Display DTI': display_dti,
+    #'Display DTI': display_dti,
     'Retrieve Top-k': retrieval
 }