Ricdeq commited on
Commit
cfe0fae
1 Parent(s): fdebbb0

req change

Browse files
Files changed (2) hide show
  1. requirements.txt +2 -0
  2. setup.py +3 -0
requirements.txt CHANGED
@@ -1,2 +1,4 @@
1
  gradio
2
  torch
 
 
 
1
  gradio
2
  torch
3
+ torchvision
4
+ Pillow
setup.py CHANGED
@@ -7,7 +7,10 @@ setup(
7
  install_requires=[
8
  'gradio',
9
  'torch',
 
 
10
  ],
 
11
  entry_points={
12
  'console_scripts': [
13
  'analyze_blueprint=blueprint_analyzer.analyze_blueprint:main',
 
7
  install_requires=[
8
  'gradio',
9
  'torch',
10
+ 'torchvision',
11
+ 'Pillow',
12
  ],
13
+
14
  entry_points={
15
  'console_scripts': [
16
  'analyze_blueprint=blueprint_analyzer.analyze_blueprint:main',