tensorkelechi commited on
Commit
6010aaa
·
verified ·
1 Parent(s): c0b1cbd

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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=config.image_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)