callmeclover commited on
Commit
af2bfeb
1 Parent(s): 2e9b607

Rename app.py to setup.py

Browse files
Files changed (2) hide show
  1. app.py +0 -9
  2. setup.py +61 -0
app.py DELETED
@@ -1,9 +0,0 @@
1
- from transformers import BlenderbotTokenizer, BlenderbotForConditionalGeneration
2
-
3
- mname = "facebook/blenderbot-400M-distill"
4
- model = BlenderbotForConditionalGeneration.from_pretrained(mname)
5
- tokenizer = BlenderbotTokenizer.from_pretrained(mname)
6
- UTTERANCE = "My friends are cool but they eat too many carbs."
7
- inputs = tokenizer([UTTERANCE], return_tensors="pt")
8
- reply_ids = model.generate(**inputs)
9
- print(tokenizer.batch_decode(reply_ids))
 
 
 
 
 
 
 
 
 
 
setup.py ADDED
@@ -0,0 +1,61 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import pathlib
2
+ import setuptools
3
+ import pkg_resources
4
+
5
+
6
+ with open("README.md", "r") as fh:
7
+ long_description = fh.read()
8
+
9
+ with pathlib.Path('requirements.txt').open() as requirements_txt:
10
+ install_requires = [
11
+ str(requirement)
12
+ for requirement
13
+ in pkg_resources.parse_requirements(requirements_txt)
14
+ ]
15
+
16
+ setuptools.setup(
17
+ name="python-transformer-chatbots-butyr",
18
+ version="0.1.2",
19
+ package_dir={"": "src"},
20
+ packages=setuptools.find_namespace_packages(where="src"),
21
+ author="Leonid Butyrev",
22
+ author_email="L.Butyrev@gmx.de",
23
+ description="A collection of transformer based chatbots.",
24
+ long_description=long_description,
25
+ long_description_content_type="text/markdown",
26
+ url="https://github.com/butyr/python-transformer-chatbot",
27
+ classifiers=[
28
+ "Programming Language :: Python :: 3",
29
+ "License :: OSI Approved :: GPL 2.0 License",
30
+ "Operating System :: OS Independent",
31
+ ],
32
+ python_requires='>=3.7',
33
+ install_requires=[
34
+ 'certifi==2021.5.30',
35
+ 'charset-normalizer==2.0.6',
36
+ 'click==8.0.1',
37
+ 'filelock==3.1.0',
38
+ 'huggingface-hub==0.0.17',
39
+ 'idna==3.2',
40
+ 'importlib-metadata==4.8.1',
41
+ 'joblib==1.0.1',
42
+ 'numpy==1.21.2',
43
+ 'packaging==21.0',
44
+ 'Pillow==9.0.1',
45
+ 'pyparsing==2.4.7',
46
+ 'PyYAML==5.4.1',
47
+ 'regex==2021.9.24',
48
+ 'requests==2.26.0',
49
+ 'sacremoses==0.0.46',
50
+ 'six==1.16.0',
51
+ 'tokenizers==0.10.3',
52
+ 'torch==1.9.1',
53
+ 'torchaudio==0.9.1',
54
+ 'torchvision==0.10.1',
55
+ 'tqdm==4.62.3',
56
+ 'transformers==4.11.0',
57
+ 'typing-extensions==3.10.0.2',
58
+ 'urllib3==1.26.7',
59
+ 'zipp==3.5.0',
60
+ ]
61
+ )