mshukor HF Staff commited on
Commit
30f9bc3
·
1 Parent(s): eff7d35

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -115,11 +115,11 @@ def read_video(path, num_frames=16):
115
  )
116
  video = test_transform(frames)
117
 
118
- return video
119
 
120
 
121
  do_sample=False
122
- num_beams=3
123
  max_length=30
124
 
125
 
 
115
  )
116
  video = test_transform(frames)
117
 
118
+ return video.unsqueeze(0)
119
 
120
 
121
  do_sample=False
122
+ num_beams=5
123
  max_length=30
124
 
125