From e93e5a0c16f75b8519a0560f76ff1cf116fd5df6 Mon Sep 17 00:00:00 2001
From: James Betker <jbetker@gmail.com>
Date: Mon, 2 May 2022 21:24:34 -0600
Subject: [PATCH] Add setup

---
 README.md        |  7 +++++--
 requirements.txt |  3 +--
 setup.py         | 36 ++++++++++++++++++++++++++++++++++++
 3 files changed, 42 insertions(+), 4 deletions(-)
 create mode 100644 setup.py

diff --git a/README.md b/README.md
index 8aef3a0..cabf457 100644
--- a/README.md
+++ b/README.md
@@ -35,12 +35,15 @@ https://colab.research.google.com/drive/1wVVqUPqwiDBUVeWWOUNglpGhU3hg_cbR?usp=sh
 
 ### Installation
 
-If you want to use this on your own computer, you must have an NVIDIA GPU. Installation:
+If you want to use this on your own computer, you must have an NVIDIA GPU. First, install pytorch using these
+instructions: [https://pytorch.org/get-started/locally/](https://pytorch.org/get-started/locally/)
+
+Then:
 
 ```shell
 git clone https://github.com/neonbjb/tortoise-tts.git
 cd tortoise-tts
-pip install -r requirements.txt
+python setup.py install
 ```
 
 ### do_tts.py
diff --git a/requirements.txt b/requirements.txt
index 91503ea..85d0f26 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,5 +1,4 @@
-torch
-torchaudio
+tqdm
 rotary_embedding_torch
 transformers
 tokenizers
diff --git a/setup.py b/setup.py
new file mode 100644
index 0000000..fa5be35
--- /dev/null
+++ b/setup.py
@@ -0,0 +1,36 @@
+import setuptools
+
+with open("README.md", "r", encoding="utf-8") as fh:
+    long_description = fh.read()
+
+setuptools.setup(
+    name="TorToiSe",
+    packages=["tortoise"],
+    version="2.1.0",
+    author="James Betker",
+    author_email="james@adamant.ai",
+    description="A high quality multi-voice text-to-speech library",
+    long_description=long_description,
+    long_description_content_type="text/markdown",
+    url="https://github.com/neonbjb/tortoise-tts",
+    project_urls={},
+    install_requires=[
+        'tqdm',
+        'rotary_embedding_torch',
+        'inflect',
+        'progressbar',
+        'einops',
+        'unidecode',
+        'entmax',
+        'scipy',
+        'librosa',
+        'transformers',
+        'tokenizers',
+    ],
+    classifiers=[
+        "Programming Language :: Python :: 3",
+        "License :: OSI Approved :: Apache Software License",
+        "Operating System :: OS Independent",
+    ],
+    python_requires=">=3.6",
+)
\ No newline at end of file