Update script.py

script.py

@@ -26,7 +26,7 @@ def predict(test_metadata, root_path='/tmp/data/private_testset', output_csv_pat
     for img_name in img_name_list:
         img_path = os.path.join(root_path, img_name)
         assert os.path.exists(img_path), "file: '{}' dose not exist.".format(img_path)
-        img = Image.open(img_path)
+        img = Image.open(img_path).convert('RGB')
         img = data_transform(img)
         img_list.append(img)