j-min commited on
Commit
1774e89
Β·
1 Parent(s): 46711d5

Simplify requirements

Browse files
Files changed (2) hide show
  1. app.py +1 -4
  2. requirements.txt +2 -26
app.py CHANGED
@@ -16,7 +16,6 @@ from datetime import datetime
16
 
17
  import torch
18
  from PIL import Image
19
- import matplotlib.pyplot as plt
20
  from diffusers import StableDiffusionInpaintPipeline
21
  from accelerate.utils import set_seed
22
 
@@ -48,7 +47,7 @@ class Instance:
48
 
49
  if torch.cuda.is_available():
50
  pipe = pipe.to('cuda')
51
-
52
  def dummy(images, **kwargs):
53
  return images, False
54
  pipe.safety_checker = dummy
@@ -721,8 +720,6 @@ with Blocks(
721
  cache_examples=False,
722
  )
723
 
724
- # https://huggingface.co/spaces/gligen/demo
725
- # add hyperlink
726
  thank_desc = """
727
  Thanks
728
  <a href="https://huggingface.co/spaces/gligen/demo" target="_blank">GLIGEN demo</a>, for providing bounding box parsing module.
 
16
 
17
  import torch
18
  from PIL import Image
 
19
  from diffusers import StableDiffusionInpaintPipeline
20
  from accelerate.utils import set_seed
21
 
 
47
 
48
  if torch.cuda.is_available():
49
  pipe = pipe.to('cuda')
50
+
51
  def dummy(images, **kwargs):
52
  return images, False
53
  pipe.safety_checker = dummy
 
720
  cache_examples=False,
721
  )
722
 
 
 
723
  thank_desc = """
724
  Thanks
725
  <a href="https://huggingface.co/spaces/gligen/demo" target="_blank">GLIGEN demo</a>, for providing bounding box parsing module.
requirements.txt CHANGED
@@ -2,31 +2,7 @@
2
  transformers
3
  diffusers
4
  accelerate
5
- opencv-python
6
- timm
7
- einops
8
- datasets
9
- rouge_score
10
- omegaconf
11
- ftfy
12
- pycocotools
13
- pycocoevalcap
14
- albumentations
15
- pudb
16
  imageio
17
  imageio-ffmpeg
18
- clean-fid
19
- h5py
20
- pillow
21
- setuptools
22
- pytorch-lightning==1.5.9
23
- tensorboardX==2.4.1
24
- test-tube>=0.7.5
25
- streamlit>=0.73.1
26
- torch-fidelity==0.3.0
27
- torchmetrics==0.6.0
28
- kornia==0.6
29
- chardet
30
- cchardet
31
- taming-transformers-rom1504
32
- git+https://github.com/openai/CLIP.git
 
2
  transformers
3
  diffusers
4
  accelerate
5
+ numpy
 
 
 
 
 
 
 
 
 
 
6
  imageio
7
  imageio-ffmpeg
8
+ pillow