emmas96 commited on
Commit
af1becc
·
1 Parent(s): 35aa55d

remove temporary prints

Browse files
Files changed (1) hide show
  1. app.py +2 -4
app.py CHANGED
@@ -27,7 +27,7 @@ st.title('HyperDTI: Robust Task-Conditioned Modeling of Drug-Target Interactions
27
  st.markdown('')
28
  st.markdown(
29
  """
30
- 🧬 Github: [ml-jku/hyper-dti](https://https://github.com/ml-jku/hyper-dti) 📝 NeurIPS 2022 AI4Science workshop paper: [OpenReview](https://openreview.net/forum?id=dIX34JWnIAL) TBA Journal of Chemical Information and Modeling. \n
31
  """
32
  )
33
 
@@ -86,9 +86,7 @@ def retrieval():
86
  with open(os.path.join(data_path, f'Lenselink/processed/SeqVec_encoding_test.pickle'), 'rb') as handle:
87
  test_set = pickle.load(handle)
88
 
89
- print(sequence in test_set.keys())
90
- print(sequence in test_set.keys().values)
91
- if sequence in test_set.keys():
92
  query_embedding = test_set[sequence]
93
  else:
94
  from bio_embeddings.embed import SeqVecEmbedder
 
27
  st.markdown('')
28
  st.markdown(
29
  """
30
+ 🧬 Github: [ml-jku/hyper-dti](https://https://github.com/ml-jku/hyper-dti) 📝 NeurIPS 2022 AI4Science workshop paper: [OpenReview](https://openreview.net/forum?id=dIX34JWnIAL); TBA in JCIM. \n
31
  """
32
  )
33
 
 
86
  with open(os.path.join(data_path, f'Lenselink/processed/SeqVec_encoding_test.pickle'), 'rb') as handle:
87
  test_set = pickle.load(handle)
88
 
89
+ if sequence in list(test_set.keys()):
 
 
90
  query_embedding = test_set[sequence]
91
  else:
92
  from bio_embeddings.embed import SeqVecEmbedder