CognitivessAI
/

cognitivess

Text Generation

text-generation-inference

Sentiment Analysis

Inference Endpoints

8-bit precision

Model card Files Files and versions Community

cognitivess commited on Jul 20

Commit

4bd5e51

•

1 Parent(s): 10bad3c

Update setup.py

Files changed (1) hide show

setup.py +5 -4

setup.py CHANGED Viewed

@@ -1,10 +1,11 @@
 from setuptools import setup, find_packages
 setup(
-    name='cognitivess_model',
-    version='0.1',
     packages=find_packages(),
     install_requires=[
-        'transformers',
     ],
-)

 from setuptools import setup, find_packages
 setup(
+    name="cognitivess_model",
+    version="0.1",
     packages=find_packages(),
     install_requires=[
+        "transformers",
+        "torch",
     ],
+)