Download models

Author: TheNeodev

Advanced-RVC.ipynb

@@ -157,6 +157,67 @@
       "execution_count": null,
       "outputs": []
     },
+    {
+      "cell_type": "code",
+      "source": [
+        "#@title Download models\n",
+        "\n",
+        "import os\n",
+        "import requests\n",
+        "from tqdm import tqdm\n",
+        "\n",
+        "def download_model(url, model_name):\n",
+        "  \"\"\"Downloads a model from a URL to a specified directory.\n",
+        "\n",
+        "  Args:\n",
+        "    url: The URL of the model to download.\n",
+        "    model_name: The name of the model.\n",
+        "  \"\"\"\n",
+        "\n",
+        "  model_dir = os.path.join(\"models\", model_name)\n",
+        "  os.makedirs(model_dir, exist_ok=True)\n",
+        "  zip_path = os.path.join(model_dir, \"model.zip\")\n",
+        "\n",
+        "  try:\n",
+        "      response = requests.get(url, stream=True)\n",
+        "      response.raise_for_status() # Raise an exception for bad status codes\n",
+        "\n",
+        "      total_size = int(response.headers.get('content-length', 0))\n",
+        "      block_size = 1024 # 1 KB\n",
+        "      progress_bar = tqdm(total=total_size, unit='iB', unit_scale=True)\n",
+        "\n",
+        "      with open(zip_path, 'wb') as file:\n",
+        "          for data in response.iter_content(block_size):\n",
+        "              progress_bar.update(len(data))\n",
+        "              file.write(data)\n",
+        "      progress_bar.close()\n",
+        "\n",
+        "      if total_size != 0 and progress_bar.n != total_size:\n",
+        "          print(\"ERROR, something went wrong\")\n",
+        "\n",
+        "      print(f\"Model '{model_name}' downloaded successfully to {model_dir}\")\n",
+        "\n",
+        "  except requests.exceptions.RequestException as e:\n",
+        "      print(f\"Error downloading model: {e}\")\n",
+        "  except Exception as e:\n",
+        "      print(f\"An unexpected error occurred: {e}\")\n",
+        "\n",
+        "\n",
+        "# Example usage\n",
+        "model_url = \"\" #@param {type:\"string\"}\n",
+        "model_name = \"\"  #@param {type:\"string\"}\n",
+        "# Replace with your desired model name\n",
+        "\n",
+        "\n",
+        "download_model(model_url, model_name)"
+      ],
+      "metadata": {
+        "cellView": "form",
+        "id": "qk74gqJqEB_A"
+      },
+      "execution_count": null,
+      "outputs": []
+    },
     {
       "cell_type": "code",
       "source": [