From e4824f7d992e4730af2b3e0fefd4837f641b25bc Mon Sep 17 00:00:00 2001
From: Josh Vendrow <jvendrow@gmail.com>
Date: Mon, 27 Feb 2023 15:25:19 -0500
Subject: [PATCH] updates for pypi upload

---
 README.md                      | 5 ++++-
 dataset_interfaces/__init__.py | 4 +++-
 setup.py                       | 6 +++---
 3 files changed, 10 insertions(+), 5 deletions(-)

diff --git a/README.md b/README.md
index ac2e64c..56bf1ca 100644
--- a/README.md
+++ b/README.md
@@ -12,7 +12,10 @@ Blog post: [https://gradientscience.org/dataset-interfaces/](https://gradientsci
 </p>
 
 ## Getting started
-To begin, clone our repository.
+Install using pip, or clone our repository.
+```
+pip install dataset-interfaces
+```
 
 **Example:** For a walkthrough of codebase, check out our [example notebook](notebooks/Example.ipynb). This notebook shows how to
 construct a dataset interface for a subset of ImageNet and generate counterfactual examples. 
diff --git a/dataset_interfaces/__init__.py b/dataset_interfaces/__init__.py
index 16b27a4..4c2d68b 100644
--- a/dataset_interfaces/__init__.py
+++ b/dataset_interfaces/__init__.py
@@ -1,2 +1,4 @@
+__version__ = "0.1.0"
+
 from .textual_inversion import run_textual_inversion
-from .generate import generate
\ No newline at end of file
+from .generate import generate
diff --git a/setup.py b/setup.py
index 8624e81..89cfcd4 100644
--- a/setup.py
+++ b/setup.py
@@ -8,9 +8,9 @@
 
 setup(
     name='dataset_interfaces',
-    version='0.0.1',    
+    version='0.1.0',    
     description='Dataset Interfaces',
-    url='https://github.com/MadryLab/dataset-interfacees',
+    url='https://github.com/MadryLab/dataset-interfaces',
     author='MadryLab',
     author_email='jvendrow@mit.edu',
     license = 'MIT',
@@ -22,7 +22,7 @@
                       'numpy',  
                       'torchvision',
                       'diffusers',
-                      'transformers'
+                      'transformers',
                       'accelerate'
                       ],
 )