Spaces:
Runtime error
Runtime error
syntax fix
Browse files
model.py
CHANGED
@@ -50,7 +50,7 @@ class ImageLoader():
|
|
50 |
x,y = image.shape[-2:]
|
51 |
ratio = float(self.show_size/max((x,y)))
|
52 |
image = torchvision.transforms.functional.resize(image, [int(x * ratio), int(y * ratio)])
|
53 |
-
return
|
54 |
|
55 |
|
56 |
class VirTexModel():
|
50 |
x,y = image.shape[-2:]
|
51 |
ratio = float(self.show_size/max((x,y)))
|
52 |
image = torchvision.transforms.functional.resize(image, [int(x * ratio), int(y * ratio)])
|
53 |
+
return torchvision.transforms.functional.to_pil_image(image)
|
54 |
|
55 |
|
56 |
class VirTexModel():
|