osanseviero HF staff commited on
Commit
6ef6b1d
1 Parent(s): 106b6ac

Update pipeline.py

Browse files
Files changed (1) hide show
  1. pipeline.py +2 -0
pipeline.py CHANGED
@@ -2,6 +2,8 @@ import torch
2
  import nltk
3
  import io
4
  import base64
 
 
5
 
6
  from pytorch_pretrained_biggan import BigGAN, one_hot_from_names, truncated_noise_sample
7
 
 
2
  import nltk
3
  import io
4
  import base64
5
+ from torchvision import transforms
6
+
7
 
8
  from pytorch_pretrained_biggan import BigGAN, one_hot_from_names, truncated_noise_sample
9