osanseviero
commited on
Commit
·
6ef6b1d
1
Parent(s):
106b6ac
Update pipeline.py
Browse files- 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 |
|