Hugging Face's logo

Spaces:
orrzohar
/
Video-STaR
Sleeping

App Files Community
Orr Zohar commited on
Commit
1ffe4d1
·
1 Parent(s): 1fbdd31

decrease batch size

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -16,7 +16,7 @@ LABEL = 'label'
16
  PREDICTION = 'prediction'
17
  left_side_columns = [VIDEO_NAME]
18
  right_side_columns = [QUESTION, LABEL, PREDICTION]
19
- batch_size = 8
20
  target_size = (1024, 1024)
21
 
22
  def func(index, dataset):
 
16
  PREDICTION = 'prediction'
17
  left_side_columns = [VIDEO_NAME]
18
  right_side_columns = [QUESTION, LABEL, PREDICTION]
19
+ batch_size = 4
20
  target_size = (1024, 1024)
21
 
22
  def func(index, dataset):