diff --git a/setup.py b/setup.py
index b9dd776..207fc50 100644
--- a/setup.py
+++ b/setup.py
@@ -24,8 +24,8 @@ def read(fname):
         "stringdist"
     ],
     extras_require = {
-        'HF': '',
-        'torch': ''
+        'huggingface': 'datasets',
+        'torch': 'torch'
     }
     long_description=read("README.md"),
     long_description_content_type='text/markdown'