fix app.py
Browse files
app.py
CHANGED
|
@@ -402,7 +402,7 @@ for p in caption_encoder.parameters():
|
|
| 402 |
# Классификационная голова
|
| 403 |
model = ConcatFusionModel(NUM_CLASSES, dropout=0.3)
|
| 404 |
model.load_state_dict(torch.load(os.path.join(
|
| 405 |
-
BASE_DIR, "
|
| 406 |
model.to(DEVICE)
|
| 407 |
model.eval()
|
| 408 |
|
|
|
|
| 402 |
# Классификационная голова
|
| 403 |
model = ConcatFusionModel(NUM_CLASSES, dropout=0.3)
|
| 404 |
model.load_state_dict(torch.load(os.path.join(
|
| 405 |
+
BASE_DIR, "concat_model.pth"), map_location=DEVICE))
|
| 406 |
model.to(DEVICE)
|
| 407 |
model.eval()
|
| 408 |
|