Hugging Face's logo

Spaces:
ctran
/
VideoMamba
Runtime error

App Files Community
SakuraD commited on
Commit
11b4d2f
•
1 Parent(s): c71518c

fix bug

Browse files
Files changed (2) hide show
  1. app.py +2 -2
  2. requirements.txt +2 -4
app.py CHANGED
@@ -75,8 +75,8 @@ def load_video(video_path):
75
  frame_indices = get_index(num_frames, 16)
76
 
77
  # transform
78
- crop_size = 160
79
- scale_size = 160
80
  input_mean = [0.485, 0.456, 0.406]
81
  input_std = [0.229, 0.224, 0.225]
82
 
 
75
  frame_indices = get_index(num_frames, 16)
76
 
77
  # transform
78
+ crop_size = 224
79
+ scale_size = 224
80
  input_mean = [0.485, 0.456, 0.406]
81
  input_std = [0.229, 0.224, 0.225]
82
 
requirements.txt CHANGED
@@ -8,7 +8,5 @@ setuptools
8
  timm
9
  transformers
10
  wheel
11
- torch
12
- torchvision
13
- # torch==2.1.1+cu118 --find-links https://download.pytorch.org/whl/cu118
14
- # torchvision==0.16.1+cu118 --find-links https://download.pytorch.org/whl/cu118
 
8
  timm
9
  transformers
10
  wheel
11
+ torch --extra-index-url https://download.pytorch.org/whl/cu118
12
+ torchvision --extra-index-url https://download.pytorch.org/whl/cu118