PeterYoung777 commited on
Commit
488377e
1 Parent(s): b0c8de1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -16,7 +16,7 @@ def predict(img):
16
  img_size = {"s": [300, 384], # train_size, val_size
17
  "m": [384, 480],
18
  "l": [384, 480]}
19
- num_model = "s"
20
 
21
  data_transform = transforms.Compose(
22
  [transforms.Resize(img_size[num_model][1]),
 
16
  img_size = {"s": [300, 384], # train_size, val_size
17
  "m": [384, 480],
18
  "l": [384, 480]}
19
+ num_model = "m"
20
 
21
  data_transform = transforms.Compose(
22
  [transforms.Resize(img_size[num_model][1]),