diff --git a/demo_part1.ipynb b/demo_part1.ipynb
index 225b7d3..81484f8 100644
--- a/demo_part1.ipynb
+++ b/demo_part1.ipynb
@@ -38,7 +38,7 @@
    "source": [
     "ckpt_base = 'checkpoints/base_speakers/EN'\n",
     "ckpt_converter = 'checkpoints/converter'\n",
-    "device = 'cuda:0'\n",
+    "device=\"cuda:0\" if torch.cuda.is_available() else \"cpu\"\n",
     "output_dir = 'outputs'\n",
     "\n",
     "base_speaker_tts = BaseSpeakerTTS(f'{ckpt_base}/config.json', device=device)\n",
@@ -214,7 +214,7 @@
    "hash": "9d70c38e1c0b038dbdffdaa4f8bfa1f6767c43760905c87a9fbe7800d18c6c35"
   },
   "kernelspec": {
-   "display_name": "Python 3.9.18 ('openvoice')",
+   "display_name": "Python 3 (ipykernel)",
    "language": "python",
    "name": "python3"
   },
diff --git a/demo_part2.ipynb b/demo_part2.ipynb
index ac41c8a..1ce6aa2 100644
--- a/demo_part2.ipynb
+++ b/demo_part2.ipynb
@@ -37,7 +37,7 @@
    "outputs": [],
    "source": [
     "ckpt_converter = 'checkpoints/converter'\n",
-    "device = 'cuda:0'\n",
+    "device=\"cuda:0\" if torch.cuda.is_available() else \"cpu\"\n",
     "output_dir = 'outputs'\n",
     "\n",
     "tone_color_converter = ToneColorConverter(f'{ckpt_converter}/config.json', device=device)\n",
@@ -173,7 +173,7 @@
    "hash": "9d70c38e1c0b038dbdffdaa4f8bfa1f6767c43760905c87a9fbe7800d18c6c35"
   },
   "kernelspec": {
-   "display_name": "Python 3.9.18 ('openvoice')",
+   "display_name": "Python 3 (ipykernel)",
    "language": "python",
    "name": "python3"
   },