impresso_langident_wrapper.py

import floret  # Assuming Floret is already installed


class FloretLangIdentifier:
    def __init__(self, model_path):
        self.model = floret.load_model(model_path)

    def predict(self, text):
        predictions = self.model.predict(text)
        return predictions