Someshfengde
commited on
Commit
•
0009a2b
1
Parent(s):
1c610d5
Upload folder using huggingface_hub
Browse files
script.py
CHANGED
@@ -60,9 +60,8 @@ def make_submission(test_metadata, model_path, model_name, output_csv_path="./su
|
|
60 |
model = PytorchWorker(model_path, model_name)
|
61 |
|
62 |
predictions = []
|
63 |
-
|
64 |
for _, row in tqdm(test_metadata.iterrows(), total=len(test_metadata)):
|
65 |
-
image_path = os.path.join(images_root_path, row.
|
66 |
|
67 |
test_image = Image.open(image_path).convert("RGB")
|
68 |
|
|
|
60 |
model = PytorchWorker(model_path, model_name)
|
61 |
|
62 |
predictions = []
|
|
|
63 |
for _, row in tqdm(test_metadata.iterrows(), total=len(test_metadata)):
|
64 |
+
image_path = os.path.join(images_root_path, row.filename)
|
65 |
|
66 |
test_image = Image.open(image_path).convert("RGB")
|
67 |
|