Wwwy1031 commited on
Commit
bfb576f
·
verified ·
1 Parent(s): 8577905

Update avp_web/engine.py

Browse files
Files changed (1) hide show
  1. 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.AVP_HNCL_v3(esm_sequence_embedding, additional_features)
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 AVP_HNCL_v3
128
 
129
- model = AVP_HNCL_v3(
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,