main_func.py

import joblib
from DeepDR import Data, Model


def func2(cell_name, drug_name, model_file, cell_input, drug_input):
    model = joblib.load(model_file.name)
    data = Data.DrData(pair_ls=[[cell_name, drug_name], [cell_name, drug_name]], cell_ft=cell_input, drug_ft=drug_input)
    result = Model.Predict(model=model, data=data)
    return result[0][0]


def clear_func2():
    return None