sms-transaction-extractor / load_model.py
rawsun00001's picture
Upload folder using huggingface_hub
d95ab75 verified
import pickle
import json
def load_sms_extractor():
"""Load the SMS Transaction Extractor"""
return pickle.load(open("extractor.pkl", "rb"))
def extract_from_sms(sms_text):
"""Extract transaction data from SMS"""
extractor = load_sms_extractor()
return extractor.extract(sms_text)
# Example usage
if __name__ == "__main__":
sms = "A/c 2XXXXX9311 credited by Rs. 2500 on 15-08-2025"
result = extract_from_sms(sms)
print(json.dumps(result, indent=2))