Someshfengde commited on
Commit
0009a2b
1 Parent(s): 1c610d5

Upload folder using huggingface_hub

Browse files
Files changed (1) hide show
  1. script.py +1 -2
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.image_path)
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