Spaces:

Andy1621
/

uniformer_video_demo

Running

Andy1621 commited on Feb 4, 2022

Commit

639ca03

1 Parent(s): 8f49d2e

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -23,7 +23,7 @@ os.system("wget https://cdn-lfs.huggingface.co/Andy1621/uniformer/d5fd7b0c49ee6a
 # Pick a pretrained model
 model = uniformer_small()
 state_dict = torch.load('d5fd7b0c49ee6a5422ef5d0c884d962c742003bfbd900747485eb99fa269d0db', map_location='cpu')
-model.load_state_dict(state_dict['model'])
 # Set to eval mode and move to desired device
 model = model.to(device)

 # Pick a pretrained model
 model = uniformer_small()
 state_dict = torch.load('d5fd7b0c49ee6a5422ef5d0c884d962c742003bfbd900747485eb99fa269d0db', map_location='cpu')
+model.load_state_dict(state_dict)
 # Set to eval mode and move to desired device
 model = model.to(device)