Spaces:
Running
Running
Orr Zohar
commited on
Commit
•
1ffe4d1
1
Parent(s):
1fbdd31
decrease batch size
Browse files
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 =
|
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):
|