Commit
·
af7ada6
1
Parent(s):
8298f98
Update pipeline.py
Browse files- pipeline.py +1 -1
pipeline.py
CHANGED
|
@@ -34,6 +34,6 @@ class PreTrainedPipeline():
|
|
| 34 |
|
| 35 |
# Ugly hack, I'm sure there's a better way of doing what I want.
|
| 36 |
img = output[0]
|
| 37 |
-
img = (img + 1) / 2.0
|
| 38 |
img = transforms.ToPILImage()(img)
|
| 39 |
return img
|
|
|
|
| 34 |
|
| 35 |
# Ugly hack, I'm sure there's a better way of doing what I want.
|
| 36 |
img = output[0]
|
| 37 |
+
img = (img + 1) / 2.0
|
| 38 |
img = transforms.ToPILImage()(img)
|
| 39 |
return img
|