Update app.py
Browse files
app.py
CHANGED
@@ -101,11 +101,11 @@ def resize_image(image):
|
|
101 |
|
102 |
|
103 |
def process(image):
|
104 |
-
return image
|
105 |
# prepare input
|
106 |
-
|
107 |
-
|
108 |
-
|
|
|
109 |
# im_np = np.array(image)
|
110 |
# im_tensor = torch.tensor(im_np, dtype=torch.float32).permute(2,0,1)
|
111 |
# im_tensor = torch.unsqueeze(im_tensor,0)
|
|
|
101 |
|
102 |
|
103 |
def process(image):
|
|
|
104 |
# prepare input
|
105 |
+
orig_image = image
|
106 |
+
w,h = orig_im_size = orig_image.size
|
107 |
+
image = resize_image(orig_image)
|
108 |
+
return image;
|
109 |
# im_np = np.array(image)
|
110 |
# im_tensor = torch.tensor(im_np, dtype=torch.float32).permute(2,0,1)
|
111 |
# im_tensor = torch.unsqueeze(im_tensor,0)
|