{
 "cells": [
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "import threading\n",
    "import yaml\n",
    "\n",
    "with open(\"/content/IAekovnI/db/configs/models.yaml\", \"r\") as f:\n",
    "    models = yaml.safe_load(f)\n",
    "\n",
    "new_models = {\n",
    "    \"sd-1/main/chilloutmix_NiPrunedFp32Fix-inpainting.inpainting\": {\n",
    "        \"path\": \"/content/IAekovnI/db/models/sd-1/main/chilloutmix_NiPrunedFp32Fix-inpainting.inpainting.safetensors\",\n",
    "        \"config\": \"/content/IAekovnI/db/configs/stable-diffusion/v1-inpainting-inference.yaml\",\n",
    "        \"variant\": \"inpaint\",\n",
    "        \"format\": \"checkpoint\",\n",
    "    },\n",
    "}\n",
    "\n",
    "if \"sd-1/main/chilloutmix_NiPrunedFp32Fix-inpainting.inpainting\" not in models:\n",
    "    models.update(new_models)\n",
    "\n",
    "with open(\"/content/IAekovnI/db/configs/models.yaml\", \"w\") as f:\n",
    "    yaml.safe_dump(models, f)\n",
    "\n",
    "def start_tunnle():\n",
    "    if use_ngrok:\n",
    "        from pyngrok import ngrok\n",
    "        ngrok_tunnel = ngrok.connect(9090, \"http\")\n",
    "        print(\"ngrok_tunnel:\", ngrok_tunnel)\n",
    "    if use_cloudflare:\n",
    "        from pycloudflared import try_cloudflare\n",
    "        cloudflare_url = try_cloudflare(9090, verbose=False)\n",
    "        print(\"cloudflare_tunnel:\", cloudflare_url)\n",
    "\n",
    "threading.Thread(target=start_tunnle, daemon=True).start()\n",
    "\n",
    "!python /content/IAekovnI/scripts/invokeai-web.py --root /content/IAekovnI/db"
   ]
  }
 ],
 "metadata": {
  "language_info": {
   "name": "python"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 2
}