model.py

class DummyModel:
    def __init__(self):
        self.label = "dummy"

    def predict(self, text: str) -> str:
        return f"Prediction from dummy model on input: {text}"


if __name__ == "__main__":
    model = DummyModel()
    print(model.predict("Hello, Hugging Face!"))