Spaces:
Runtime error
Runtime error
| from transformers import RobertaTokenizerFast, RobertaForSequenceClassification | |
| from utils.commons import MODEL_SAVE_PATH, setup_logging | |
| logger = setup_logging("model_loader.log") | |
| class ModelLoader: | |
| def __init__(self): | |
| self.tokenizer = None | |
| self.model = None | |
| def load_model(self): | |
| """Load model from local path""" | |
| try: | |
| logger.info("Loading tokenizer and model...") | |
| self.tokenizer = RobertaTokenizerFast.from_pretrained(MODEL_SAVE_PATH) | |
| self.model = RobertaForSequenceClassification.from_pretrained(MODEL_SAVE_PATH) | |
| logger.info("Tokenizer and model loaded successfully.") | |
| return self.tokenizer, self.model | |
| except Exception as e: | |
| logger.error(f"Error loading model: {e}") | |
| raise |