tiansz commited on
Commit
1c974e4
1 Parent(s): 8aaf40e

Update README.md

Browse files

The sample code in the md file seems to have some problems, after this modification my side can run.

Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -24,7 +24,7 @@ from transformers import AutoImageProcessor, TimesformerForVideoClassification
24
  import numpy as np
25
  import torch
26
 
27
- video = list(np.random.randn(8, 3, 224, 224))
28
 
29
  processor = AutoImageProcessor.from_pretrained("facebook/timesformer-base-finetuned-k400")
30
  model = TimesformerForVideoClassification.from_pretrained("facebook/timesformer-base-finetuned-k400")
 
24
  import numpy as np
25
  import torch
26
 
27
+ video = list(np.random.randn(8, 3, 224, 224).astype(np.uint8))
28
 
29
  processor = AutoImageProcessor.from_pretrained("facebook/timesformer-base-finetuned-k400")
30
  model = TimesformerForVideoClassification.from_pretrained("facebook/timesformer-base-finetuned-k400")