abhishekrs4 commited on
Commit
b456e16
1 Parent(s): 8ca797e

updated app script

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -20,7 +20,7 @@ num_classes = 5
20
  image_height = 320
21
  mean_arr = np.array([[0.485, 0.456, 0.406]])
22
  std_arr = np.array([[0.229, 0.224, 0.225]])
23
- cassava_model = ResNetImageClassififer(num_classes, image_height)
24
 
25
  try:
26
  logging.info(f"loading model from {file_model_local}")
 
20
  image_height = 320
21
  mean_arr = np.array([[0.485, 0.456, 0.406]])
22
  std_arr = np.array([[0.229, 0.224, 0.225]])
23
+ cassava_model = ResNetImageClassififer(num_classes, pretrained=False)
24
 
25
  try:
26
  logging.info(f"loading model from {file_model_local}")