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)
|