{
  "cells": [
    {
      "cell_type": "code",
      "execution_count": null,
      "metadata": {
        "id": "PeEyOhUDHhzF"
      },
      "outputs": [],
      "source": [
        "import os\n",
        "import subprocess\n",
        "\n",
        "ROOT_DIR = \"/content\"\n",
        "REPO_URL = \"https://huggingface.co/spaces/Linaqruf/Animagine-XL\"\n",
        "REPO_DIR = os.path.join(ROOT_DIR, \"Animagine-XL\")\n",
        "\n",
        "def clone(url, dir, branch=None):\n",
        "    subprocess.run([\"git\", \"clone\", url, dir], check=True)\n",
        "    if branch:\n",
        "        subprocess.run([\"git\", \"checkout\", branch], cwd=dir, check=True)\n",
        "\n",
        "def install_deps(dir):\n",
        "    subprocess.run([\"pip\", \"install\", \"-r\", \"requirements.txt\"], cwd=dir, check=True)\n",
        "\n",
        "def main():\n",
        "    if not os.path.exists(REPO_DIR):\n",
        "        print(f\"Cloning Repository to {REPO_DIR}\")\n",
        "        clone(REPO_URL, REPO_DIR)\n",
        "        print(f\"Installing required python libraries\")\n",
        "        install_deps(REPO_DIR)\n",
        "        print(\"Done!\")\n",
        "\n",
        "    os.chdir(REPO_DIR)\n",
        "    !python app.py\n",
        "\n",
        "if __name__ == \"__main__\":\n",
        "    main()\n"
      ]
    }
  ],
  "metadata": {
    "accelerator": "GPU",
    "colab": {
      "machine_shape": "hm",
      "provenance": [],
      "gpuType": "A100"
    },
    "kernelspec": {
      "display_name": "Python 3",
      "name": "python3"
    },
    "language_info": {
      "name": "python"
    }
  },
  "nbformat": 4,
  "nbformat_minor": 0
}