Mayanand commited on
Commit
c783de4
1 Parent(s): f67ac78

fixed renset bug

Browse files
Files changed (1) hide show
  1. predict_image.py +1 -1
predict_image.py CHANGED
@@ -29,7 +29,7 @@ def load_model(model_name):
29
  fname = download_weights(effb0_net_url)
30
  print('loaded effnet')
31
  elif model_name == 'res18':
32
- model = ResnetModel
33
  fname = download_weights(res18_net_url)
34
  print('loaded resnet')
35
  else:
 
29
  fname = download_weights(effb0_net_url)
30
  print('loaded effnet')
31
  elif model_name == 'res18':
32
+ model = ResnetModel()
33
  fname = download_weights(res18_net_url)
34
  print('loaded resnet')
35
  else: