update requirements for latest dscript version, update gpu code, remove invalid shape check
Browse files- app.py +0 -1
- requirements.txt +1 -1
app.py
CHANGED
|
@@ -149,7 +149,6 @@ def predict(model_name, pairs_file, sequence_file, progress = gr.Progress()):
|
|
| 149 |
lm1 = lm_embed(seq1)
|
| 150 |
lm2 = lm_embed(seq2)
|
| 151 |
|
| 152 |
-
print(lm1.shape, lm2.shape, fold1.shape, fold2.shape)
|
| 153 |
interaction = model.predict(lm1, lm2, embed_foldseek = do_foldseek, f0 = fold1, f1 = fold2).item()
|
| 154 |
|
| 155 |
results.append([prot1, prot2, interaction])
|
|
|
|
| 149 |
lm1 = lm_embed(seq1)
|
| 150 |
lm2 = lm_embed(seq2)
|
| 151 |
|
|
|
|
| 152 |
interaction = model.predict(lm1, lm2, embed_foldseek = do_foldseek, f0 = fold1, f1 = fold2).item()
|
| 153 |
|
| 154 |
results.append([prot1, prot2, interaction])
|
requirements.txt
CHANGED
|
@@ -1,4 +1,4 @@
|
|
| 1 |
-
dscript
|
| 2 |
biopython
|
| 3 |
pandas
|
| 4 |
tqdm
|
|
|
|
| 1 |
+
dscript>=0.2.5
|
| 2 |
biopython
|
| 3 |
pandas
|
| 4 |
tqdm
|