ShuangLI59 commited on
Commit
2d78361
β€’
1 Parent(s): d4fe1e6
Files changed (2) hide show
  1. app.py +3 -2
  2. requirements.txt +1 -0
app.py CHANGED
@@ -8,10 +8,11 @@ Original file is located at
8
  """
9
 
10
 
11
- from PIL import Image
12
- from IPython.display import display
13
  import torch as th
14
 
 
15
  from glide_text2im.download import load_checkpoint
16
  from glide_text2im.model_creation import (
17
  create_model_and_diffusion,
 
8
  """
9
 
10
 
11
+ # from PIL import Image
12
+ # from IPython.display import display
13
  import torch as th
14
 
15
+
16
  from glide_text2im.download import load_checkpoint
17
  from glide_text2im.model_creation import (
18
  create_model_and_diffusion,
requirements.txt CHANGED
@@ -1,2 +1,3 @@
 
1
  git+https://github.com/energy-based-model/Compositional-Visual-Generation-with-Composable-Diffusion-Models-PyTorch
2
  git+https://github.com/openai/glide-text2im
 
1
+ torch
2
  git+https://github.com/energy-based-model/Compositional-Visual-Generation-with-Composable-Diffusion-Models-PyTorch
3
  git+https://github.com/openai/glide-text2im