Fix model class map referenc ein agile v2 notebook.

PiperOrigin-RevId: 691403357

chirp/projects/agile2/2_agile_modeling_v2.ipynb

@@ -20,7 +20,7 @@
         "from chirp.projects.hoplite import brutalism\n",
         "from chirp.projects.hoplite import score_functions\n",
         "from chirp.projects.hoplite import sqlite_impl\n",
-        "from chirp.projects.zoo import models\n"
+        "from chirp.projects.zoo import model_configs\n"
       ]
     },
     {
@@ -41,7 +41,7 @@
         "db = sqlite_impl.SQLiteGraphSearchDB.create(db_path)\n",
         "db_model_config = db.get_metadata('model_config')\n",
         "embed_config = db.get_metadata('embed_config')\n",
-        "model_class = models.model_class_map()[db_model_config.model_key]\n",
+        "model_class = model_configs.MODEL_CLASS_MAP[db_model_config.model_key]\n",
         "embedding_model = model_class.from_config(db_model_config.model_config)\n",
         "\n",
         "audio_filepath_loader = audio_loader.make_filepath_loader(\n",