| from huggingface_hub import hf_hub_download | |
| from fastai.vision.all import * | |
| from fastai.learner import load_learner | |
| import os | |
| import os.path as op | |
| import pickle | |
| HOME_DIR = os.getcwd() | |
| DATA_DIR = op.join(HOME_DIR, 'data') | |
| MODELS_DIR = op.join(HOME_DIR,'models') | |
| def load_model(): | |
| learn = load_learner(op.join(MODELS_DIR, "vit_tiny_patch16_224_2.pkl")) | |
| return learn | |
| # learn = load_learner(op.join(MODELS_DIR, "vit_tiny_patch16_224_2.pkl")) | |
| def load_image(image_name): | |
| img = op.join(DATA_DIR, image_name) | |
| return img | |