src/model/utils.py

import os


def use_low_resource_settings(option):
    if option == 'Yes':
        os.environ['CUDA_VISIBLE_DEVICES'] = ''
        os.environ['TFHUB_MODEL_LOAD_FORMAT'] = 'COMPRESSED'


def suppress_warnings(option):
    if option == 'Yes':
        os.environ['TF_CPP_MIN_LOG_LEVEL'] = '2'