Spaces:

Salimshakeel
/

video-summarization-backend

Sleeping

Salimshakeel commited on Jul 13

Commit

66de988

1 Parent(s): 1702d58

build fix

Files changed (1) hide show

services/model_loader.py CHANGED Viewed

@@ -14,6 +14,6 @@ def load_model(weights_path):
         fusion="add",
         pos_enc="absolute"
     ).to(DEVICE)
-    model.load_state_dict(torch.load(weights_path, map_location=DEVICE))
     model.eval
     return model

         fusion="add",
         pos_enc="absolute"
     ).to(DEVICE)
+    # model.load_state_dict(torch.load(weights_path, map_location=DEVICE))
     model.eval
     return model