Update Chatterbox_TTS_Colab_Demo.ipynb

Author: devnen

Chatterbox_TTS_Colab_Demo.ipynb

@@ -3,7 +3,8 @@
   "nbformat_minor": 0,
   "metadata": {
     "colab": {
-      "gpuType": "T4"
+      "gpuType": "T4",
+      "provenance": []
     },
     "kernelspec": {
       "name": "python3",
@@ -153,16 +154,16 @@
         "\n",
         "print(\"\\n=== Installing server requirements (prefer repo pins if present) ===\")\n",
         "if Path(\"requirements-nvidia.txt\").exists():\n",
-        "    sh(\"/content/bin/micromamba run -n cb311 pip install -U pip setuptools wheel\", check=True)\n",
-        "    sh(\"/content/bin/micromamba run -n cb311 pip install -r requirements-nvidia.txt\", check=True)\n",
+        "    sh(\"/content/bin/micromamba run -n cb311 pip install -U pip setuptools wheel\", check=False)\n",
+        "    sh(\"/content/bin/micromamba run -n cb311 pip install -r requirements-nvidia.txt\", check=False)\n",
         "else:\n",
         "    sh(\n",
         "        \"/content/bin/micromamba run -n cb311 pip install -U pip setuptools wheel && \"\n",
         "        \"/content/bin/micromamba run -n cb311 pip install \"\n",
         "        \"fastapi 'uvicorn[standard]' pyyaml soundfile librosa safetensors \"\n",
         "        \"python-multipart requests jinja2 watchdog aiofiles unidecode inflect tqdm \"\n",
         "        \"pydub audiotsm praat-parselmouth\",\n",
-        "        check=True\n",
+        "        check=False\n",
         "    )\n",
         "\n",
         "print(\"\\n=== Removing old stdout log ===\")\n",
@@ -191,22 +192,13 @@
         ")\n",
         "\n",
         "with open(LOG_STDOUT, \"w\", encoding=\"utf-8\", errors=\"replace\") as f:\n",
-        "    import select\n",
         "    shown_link = False\n",
         "    while True:\n",
-        "        # Wait briefly for stdout, but still poll port status regularly\n",
-        "        ready, _, _ = select.select([proc.stdout], [], [], 0.2)\n",
-        "        if ready:\n",
-        "            line = proc.stdout.readline()\n",
-        "            if line:\n",
-        "                print(line, end=\"\")\n",
-        "                f.write(line)\n",
-        "                f.flush()\n",
-        "            else:\n",
-        "                # EOF on pipe\n",
-        "                if proc.poll() is not None:\n",
-        "                    print(\"\\n=== Server process exited with code\", proc.returncode, \"===\")\n",
-        "                    break\n",
+        "        line = proc.stdout.readline()\n",
+        "        if line:\n",
+        "            print(line, end=\"\")\n",
+        "            f.write(line)\n",
+        "            f.flush()\n",
         "\n",
         "        if (not shown_link) and port_open():\n",
         "            shown_link = True\n",
@@ -215,6 +207,7 @@
         "            from google.colab.output import serve_kernel_port_as_window\n",
         "            serve_kernel_port_as_window(PORT)\n",
         "\n",
+        "            # Verify model load status via server endpoint\n",
         "            try:\n",
         "                mi = requests.get(f\"http://127.0.0.1:{PORT}/api/model-info\", timeout=2).json()\n",
         "                print(\"\\n/api/model-info:\", mi)\n",
@@ -223,7 +216,7 @@
         "\n",
         "        if proc.poll() is not None:\n",
         "            print(\"\\n=== Server process exited with code\", proc.returncode, \"===\")\n",
-        "            break\n"
+        "            break"
       ],
       "metadata": {
         "id": "n8DXZwq1uQEB"