Update inference.py

inference.py

@@ -1004,7 +1004,7 @@ class PeptiVersePredictor:
         if arch == "pooled":
             t_vec = self.wt_embedder.pooled(target_seq)
             b_vec = self._get_embedder(emb_tag or col).pooled(binder_str) if emb_tag else \
-                    (self.wt_embedder.pooled(binder_str) if col == "wt" else self.smiles_embedder.pooled(binder_str))
+                    (self.wt_embedder.pooled(binder_str) if col == "wt" else self.chemberta_embedder.pooled(binder_str))
             with torch.no_grad():
                 reg, logits = model(t_vec, b_vec)
         else: