vobecant
commited on
Commit
·
ad1f0d8
1
Parent(s):
2762176
Initial commit.
Browse files
app.py
CHANGED
|
@@ -151,7 +151,7 @@ def get_transformations(input_img):
|
|
| 151 |
|
| 152 |
def predict(input_img, cs_mapping):
|
| 153 |
input_img_pil = Image.open(input_img)
|
| 154 |
-
transform = get_transformations(
|
| 155 |
input_img = transform(input_img_pil)
|
| 156 |
input_img = torch.unsqueeze(input_img, 0)
|
| 157 |
|
|
|
|
| 151 |
|
| 152 |
def predict(input_img, cs_mapping):
|
| 153 |
input_img_pil = Image.open(input_img)
|
| 154 |
+
transform = get_transformations(input_img_pil)
|
| 155 |
input_img = transform(input_img_pil)
|
| 156 |
input_img = torch.unsqueeze(input_img, 0)
|
| 157 |
|