23f2001106 commited on
Commit
b710221
·
1 Parent(s): 26ae79e

Add setup.py for pip installation

Browse files
Files changed (1) hide show
  1. setup.py +17 -0
setup.py ADDED
@@ -0,0 +1,17 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ from setuptools import setup, find_packages
2
+
3
+ setup(
4
+ name="emo_detector",
5
+ version="0.1.0",
6
+ packages=find_packages(),
7
+ install_requires=[
8
+ "torch>=2.0.0",
9
+ "transformers>=4.35.0",
10
+ "sentencepiece",
11
+ ],
12
+ include_package_data=True,
13
+ description="BERT + FFNN text classifier for emotions",
14
+ author="NeuralNest05",
15
+ license="MIT",
16
+ url="https://huggingface.co/NeuralNest05/emo-detector",
17
+ )