examples/model_customization.ipynb

{
 "cells": [
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "from open_biomed.core.pipeline import InferencePipeline\n",
    "from open_biomed.data import Molecule, Text\n",
    "\n",
    "# Path relative to repo root (run from examples/ directory)\n",
    "cfg_path = \"../configs/text_based_molecule_editing/molt5.json\"\n",
    "pipeline = InferencePipeline(cfg_path)\n",
    "mol = [Molecule.from_smiles(\"CCCCC\")]\n",
    "text = [Text.from_str(\"wow\")]\n",
    "pipeline.run(mol, text)"
   ]
  }
 ],
 "metadata": {
  "language_info": {
   "name": "python"
  },
  "orig_nbformat": 4
 },
 "nbformat": 4,
 "nbformat_minor": 2
}