DLBot commited on
Commit
d2225db
·
verified ·
1 Parent(s): babc795

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -37,7 +37,7 @@ class enetv2(nn.Module):
37
  def load_models(model_files):
38
  models = []
39
  for model_f in model_files:
40
- model_f = os.path.join(model_dir, model_f)
41
  backbone = 'efficientnet-b0'
42
  model = enetv2(backbone, out_dim=5)
43
  model.load_state_dict(torch.load(model_f, map_location=lambda storage, loc: storage), strict=True)
 
37
  def load_models(model_files):
38
  models = []
39
  for model_f in model_files:
40
+ model_f = os.path.join( model_f)
41
  backbone = 'efficientnet-b0'
42
  model = enetv2(backbone, out_dim=5)
43
  model.load_state_dict(torch.load(model_f, map_location=lambda storage, loc: storage), strict=True)