iloncka/mosal-api

1from huggingface_hub import hf_hub_download2from fastai.vision.all import *3from fastai.learner import load_learner4import os5import os.path as op6import pickle7 8 9HOME_DIR = os.getcwd()10DATA_DIR =  op.join(HOME_DIR, 'data')11MODELS_DIR = op.join(HOME_DIR,'models')12 13 14def load_model():   15    learn = load_learner(op.join(MODELS_DIR, "vit_tiny_patch16_224_2.pkl"))16    return learn17 18# learn = load_learner(op.join(MODELS_DIR, "vit_tiny_patch16_224_2.pkl"))19def load_image(image_name):20    img = op.join(DATA_DIR, image_name)21    return img22