remove temporary prints
Browse files
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
|
| 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 |
-
|
| 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
|