utils/state.py

from .load_model_from_hub import load_model


class State:
    def __init__(self) -> None:
        self.model = None
        self.instructions = ""
        self.input_text = ""
        try:
            self.model = load_model()
        except Exception as e:
            raise Exception(e)

    def clear_all(self):
        model = None
        instructions = ""
        input_text = ""