Spaces:
Build error
Build error
import dotenv | |
from omegaconf import DictConfig, OmegaConf | |
from hydra import compose, initialize | |
# load environment variables from `.env` file if it exists | |
# recursively searches for `.env` in all folders starting from work dir | |
dotenv.load_dotenv(override=True) | |
# @hydra.main(version_base = None, config_path="configs/", config_name="test.yaml") | |
def handle_ir(img_path, input, output): | |
with initialize(version_base=None, config_path="configs", job_name="ir"): | |
config = compose(config_name="test") | |
# print(OmegaConf.to_yaml(cfg)) | |
# Imports can be nested inside @hydra.main to optimize tab completion | |
# https://github.com/facebookresearch/hydra/issues/934 | |
from src.ir.src import utils | |
from src.ir.src.testing_pipeline import test | |
# Applies optional utilities | |
utils.extras(config) | |
# Evaluate model | |
return test(config, input, output, img_path) | |