Spaces:
Running
Running
Update avp_web/engine.py
Browse files- avp_web/engine.py +3 -3
avp_web/engine.py
CHANGED
|
@@ -7,7 +7,7 @@ avp_web.engine (Space-ready)
|
|
| 7 |
- Physchem features via util.data.generate_features (expects FASTA FILE PATH)
|
| 8 |
- Loads scaler.pkl + columns.json from avp_web/assets
|
| 9 |
- Loads best_model.pth from repo root (or env AVP_CKPT)
|
| 10 |
-
- forward(): model_att.
|
| 11 |
"""
|
| 12 |
|
| 13 |
import os
|
|
@@ -124,9 +124,9 @@ class AVPPredictor:
|
|
| 124 |
if str(REPO_ROOT) not in sys.path:
|
| 125 |
sys.path.insert(0, str(REPO_ROOT))
|
| 126 |
|
| 127 |
-
from model_att import
|
| 128 |
|
| 129 |
-
model =
|
| 130 |
esm_dim=640,
|
| 131 |
additional_dim=self.additional_dim,
|
| 132 |
cnn_out_channels=256,
|
|
|
|
| 7 |
- Physchem features via util.data.generate_features (expects FASTA FILE PATH)
|
| 8 |
- Loads scaler.pkl + columns.json from avp_web/assets
|
| 9 |
- Loads best_model.pth from repo root (or env AVP_CKPT)
|
| 10 |
+
- forward(): model_att.AVP_Fusion(esm_sequence_embedding, additional_features)
|
| 11 |
"""
|
| 12 |
|
| 13 |
import os
|
|
|
|
| 124 |
if str(REPO_ROOT) not in sys.path:
|
| 125 |
sys.path.insert(0, str(REPO_ROOT))
|
| 126 |
|
| 127 |
+
from model_att import AVP_Fusion
|
| 128 |
|
| 129 |
+
model = AVP_Fusion(
|
| 130 |
esm_dim=640,
|
| 131 |
additional_dim=self.additional_dim,
|
| 132 |
cnn_out_channels=256,
|