Update app.py
Browse files
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(
|
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)
|