Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -13,7 +13,7 @@ from huggingface_hub import PyTorchModelHubMixin
|
|
13 |
from torchvision import models
|
14 |
|
15 |
|
16 |
-
def read_image(img, img_size=
|
17 |
img = np.array(img)
|
18 |
img = cv2.resize(img, (img_size, img_size)) # resize to mathc model input here
|
19 |
img = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
|
|
|
13 |
from torchvision import models
|
14 |
|
15 |
|
16 |
+
def read_image(img, img_size=100):
|
17 |
img = np.array(img)
|
18 |
img = cv2.resize(img, (img_size, img_size)) # resize to mathc model input here
|
19 |
img = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
|