Eduard-Sebastian Zamfir commited on
Commit
3576e92
1 Parent(s): a8463d2

fix requirements

Browse files
Files changed (2) hide show
  1. .gitignore +3 -0
  2. requirements.txt +1 -0
.gitignore CHANGED
@@ -1,3 +1,6 @@
1
  __pycache__/
2
  flagged/
3
  checkpoints
 
 
 
 
1
  __pycache__/
2
  flagged/
3
  checkpoints
4
+ *.pt
5
+ *.gif
6
+ *.pth
requirements.txt CHANGED
@@ -1,5 +1,6 @@
1
  torch
2
  numpy
 
3
  PyYAML
4
  Pillow>=6.2.2
5
  gradio==4.16.0
 
1
  torch
2
  numpy
3
+ einops
4
  PyYAML
5
  Pillow>=6.2.2
6
  gradio==4.16.0