import numpy as np # linear algebra
import pandas as pd # data processing, CSV file I/O (e.g. pd.read_csv)
from fastai.vision.all import *

model_path = "../input/models/modelo_final_v8.pkl"
modelo = load_learner(model_path)

test_dir = "../input/cassava-leaf-disease-classification/test_images/"
test_imgs = get_image_files(test_dir)

tst_dl = modelo.dls.test_dl(test_imgs)
preds, _ = modelo.tta(dl=tst_dl)
labels = np.argmax(preds, 1)

df_preds= pd.DataFrame(data=(pd.Series(test_imgs)).apply(lambda x: x.name), columns=['image_id'])
df_preds['label'] = labels

df_preds.to_csv("submission.csv", index = False)