VikTsrv commited on
Commit
dce3e8c
·
1 Parent(s): 93101e3

fix app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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, "best_concat_model.pth"), map_location=DEVICE))
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