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)
|