Lesson 2 - putting my model in production works but doesn't work :(

so after I created my model i’m trying to test an image that my model hasn’t seen. I’m following the code in this link here (https://github.com/fastai/course-v3/blob/master/nbs/dl1/lesson2-download.ipynb).

After running the code I get no errors; however, im not getting the output I want. I ran the code:
“”
pred_class,pred_idx,outputs = learn.predict(img)
pred_class
“”
and instead of getting what Jeremy got in the video " category: Black " I instead get the output " Category Tensor(0) "

Can anyone help me out with this problem?

1 Like