commit 0f707c909210a13fbe12b742387d6162b3e86082 Author: ModelHub XC Date: Thu Jun 18 09:40:18 2026 +0800 初始化项目，由ModelHub XC社区提供模型 Model: kartikaybagla/functiongemma-bank-sms-parser Source: Original Platform diff --git a/.gitattributes b/.gitattributes new file mode 100644 index 0000000..ee1e90e --- /dev/null +++ b/.gitattributes @@ -0,0 +1,40 @@ +*.7z filter=lfs diff=lfs merge=lfs -text +*.arrow filter=lfs diff=lfs merge=lfs -text +*.bin filter=lfs diff=lfs merge=lfs -text +*.bz2 filter=lfs diff=lfs merge=lfs -text +*.ckpt filter=lfs diff=lfs merge=lfs -text +*.ftz filter=lfs diff=lfs merge=lfs -text +*.gz filter=lfs diff=lfs merge=lfs -text +*.h5 filter=lfs diff=lfs merge=lfs -text +*.joblib filter=lfs diff=lfs merge=lfs -text +*.lfs.* filter=lfs diff=lfs merge=lfs -text +*.mlmodel filter=lfs diff=lfs merge=lfs -text +*.model filter=lfs diff=lfs merge=lfs -text +*.msgpack filter=lfs diff=lfs merge=lfs -text +*.npy filter=lfs diff=lfs merge=lfs -text +*.npz filter=lfs diff=lfs merge=lfs -text +*.onnx filter=lfs diff=lfs merge=lfs -text +*.ot filter=lfs diff=lfs merge=lfs -text +*.parquet filter=lfs diff=lfs merge=lfs -text +*.pb filter=lfs diff=lfs merge=lfs -text +*.pickle filter=lfs diff=lfs merge=lfs -text +*.pkl filter=lfs diff=lfs merge=lfs -text +*.pt filter=lfs diff=lfs merge=lfs -text +*.pth filter=lfs diff=lfs merge=lfs -text +*.rar filter=lfs diff=lfs merge=lfs -text +*.safetensors filter=lfs diff=lfs merge=lfs -text +saved_model/**/* filter=lfs diff=lfs merge=lfs -text +*.tar.* filter=lfs diff=lfs merge=lfs -text +*.tar filter=lfs diff=lfs merge=lfs -text +*.tflite filter=lfs diff=lfs merge=lfs -text +*.tgz filter=lfs diff=lfs merge=lfs -text +*.wasm filter=lfs diff=lfs merge=lfs -text +*.xz filter=lfs diff=lfs merge=lfs -text +*.zip filter=lfs diff=lfs merge=lfs -text +*.zst filter=lfs diff=lfs merge=lfs -text +*tfevents* filter=lfs diff=lfs merge=lfs -text +functiongemma-sms.gguf filter=lfs diff=lfs merge=lfs -text +functiongemma-270m-bank-sms-parser-f16.gguf filter=lfs diff=lfs merge=lfs -text +functiongemma-270m-bank-sms-parser-Q4_K_M.gguf filter=lfs diff=lfs merge=lfs -text +functiongemma-270m-bank-sms-parser-Q5_K_M.gguf filter=lfs diff=lfs merge=lfs -text +functiongemma-270m-bank-sms-parser-Q8_0.gguf filter=lfs diff=lfs merge=lfs -text diff --git a/README.md b/README.md new file mode 100644 index 0000000..58ee333 --- /dev/null +++ b/README.md @@ -0,0 +1,133 @@ +--- +license: gemma +base_model: google/functiongemma-270m-it +tags: + - function-calling + - finance + - sms-parsing + - transaction-extraction + - gguf + - llama-cpp +language: + - en +pipeline_tag: text-generation +library_name: transformers +--- + +# FunctionGemma Bank SMS Parser + +A fine-tuned [FunctionGemma-270M-IT](https://huggingface.co/google/functiongemma-270m-it) model for extracting structured transaction data from bank SMS messages. + +## Model Description + +This model is trained to perform two functions: + +1. **`extract_transaction`** - Parse banking SMS and extract structured fields: + - `source`: Bank or sender name + - `currency`: Currency code (INR, USD, etc.) + - `amount`: Transaction amount (number) + - `date`: Transaction date + - `destination`: Recipient or merchant + - `type`: "debit" or "credit" + +2. **`skip_message`** - Identify non-transaction messages: + - OTPs and verification codes + - Promotional messages + - Payment requests (not completed transactions) + - Account alerts without transactions + +## Quantization Options + +| File | Quantization | Size | Description | +|------|--------------|------|-------------| +| `functiongemma-270m-bank-sms-parser-Q4_K_M.gguf` | Q4_K_M | ~242MB | **Recommended** - Best size/quality tradeoff | +| `functiongemma-270m-bank-sms-parser-Q5_K_M.gguf` | Q5_K_M | ~248MB | Higher quality if Q4 shows issues | +| `functiongemma-270m-bank-sms-parser-Q8_0.gguf` | Q8_0 | ~280MB | Near-lossless, for validation | + +## Usage + +### With llama.cpp server + +```bash +# Download model +huggingface-cli download kartikaybagla/functiongemma-bank-sms-parser \ + functiongemma-270m-bank-sms-parser-Q4_K_M.gguf \ + --local-dir ./models + +# Run server +llama-server --model ./models/functiongemma-270m-bank-sms-parser-Q4_K_M.gguf \ + --host 0.0.0.0 --port 8080 --ctx-size 2048 +``` + +### With Docker + +```bash +docker run -p 8080:8080 -v ./models:/models \ + ghcr.io/ggml-org/llama.cpp:server \ + --model /models/functiongemma-270m-bank-sms-parser-Q4_K_M.gguf \ + --host 0.0.0.0 --port 8080 +``` + +### API Request + +```bash +curl http://localhost:8080/v1/completions \ + -H "Content-Type: application/json" \ + -d '{ + "prompt": "developer\nYou are a financial transaction extractor. Analyze SMS messages and:\n1. If the message describes a completed financial transaction (money sent, received, debited, or credited), use extract_transaction to capture the details.\n2. If the message is not a transaction (OTP, promotional, application status, payment request, etc.), use skip_message.\n\nOnly extract actual completed transactions with concrete amounts, not payment requests or pending transactions.declaration:extract_transaction{description:Extract transaction details from a banking SMS message,parameters:{properties:{source:{type:STRING},currency:{type:STRING},amount:{type:NUMBER},date:{type:STRING},destination:{type:STRING},type:{type:STRING}},required:[source,currency,amount,date,destination,type],type:OBJECT}}declaration:skip_message{description:Skip messages that are not financial transactions,parameters:{properties:{reason:{type:STRING}},required:[reason],type:OBJECT}}\nuser\nICICI Bank Acct XX123 debited Rs 450.00 on 15-Jan-25; UPI to SWIGGY. UPI Ref: 123456789012\nmodel\n", + "max_tokens": 200, + "stop": [""] + }' +``` + +### Example Output + +**Input SMS:** +``` +ICICI Bank Acct XX123 debited Rs 450.00 on 15-Jan-25; UPI to SWIGGY. UPI Ref: 123456789012 +``` + +**Model Output:** +``` +extract_transaction{"source": "ICICI Bank", "currency": "INR", "amount": 450.00, "date": "15-Jan-25", "destination": "SWIGGY", "type": "debit"} +``` + +**Input SMS (non-transaction):** +``` +Your OTP for login is 482910. Valid for 5 minutes. Do not share. +``` + +**Model Output:** +``` +skip_message{"reason": "OTP verification code"} +``` + +## Training + +- **Base Model**: [google/functiongemma-270m-it](https://huggingface.co/google/functiongemma-270m-it) +- **Training Framework**: Hugging Face TRL (SFTTrainer) +- **Training Data**: Classified bank SMS messages from Indian banks +- **Fine-tuning Method**: LoRA + +## Intended Use + +This model is designed for: +- Personal finance automation +- Importing transactions into budgeting apps (e.g., Actual Budget) +- SMS-based expense tracking + +## Limitations + +- Primarily trained on Indian bank SMS formats (ICICI, HDFC, SBI, etc.) +- May not generalize well to banks from other countries +- Requires the specific prompt format shown above +- Not suitable for security-critical applications without additional validation + +## License + +This model inherits the [Gemma license](https://ai.google.dev/gemma/terms) from the base model. + +## Links + +- [Project Repository](https://github.com/kartikaybagla/bank-sms-parsing) +- [Base Model](https://huggingface.co/google/functiongemma-270m-it) diff --git a/functiongemma-270m-bank-sms-parser-Q4_K_M.gguf b/functiongemma-270m-bank-sms-parser-Q4_K_M.gguf new file mode 100644 index 0000000..6e4ce27 --- /dev/null +++ b/functiongemma-270m-bank-sms-parser-Q4_K_M.gguf @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:538a672eda0fbb08ba01b217e42db6efd8f73c2a7dbd62bfdf450abb9ce3ea45 +size 253126816 diff --git a/functiongemma-270m-bank-sms-parser-Q5_K_M.gguf b/functiongemma-270m-bank-sms-parser-Q5_K_M.gguf new file mode 100644 index 0000000..3480258 --- /dev/null +++ b/functiongemma-270m-bank-sms-parser-Q5_K_M.gguf @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:14d4b12512745e1bd770c2bb1172bd674143e62ca8204157d62a5573cbb6b1b0 +size 260038816 diff --git a/functiongemma-270m-bank-sms-parser-Q8_0.gguf b/functiongemma-270m-bank-sms-parser-Q8_0.gguf new file mode 100644 index 0000000..b60b1d0 --- /dev/null +++ b/functiongemma-270m-bank-sms-parser-Q8_0.gguf @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:5a9ecbda171920b2c39968b6aa92f2c694f425d20aee7e8942c6c32205f9b94c +size 291557536 diff --git a/functiongemma-270m-bank-sms-parser-f16.gguf b/functiongemma-270m-bank-sms-parser-f16.gguf new file mode 100644 index 0000000..7e3d2c8 --- /dev/null +++ b/functiongemma-270m-bank-sms-parser-f16.gguf @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:78f8e3f47ec8e1267a8925639bdcf2fb25982e77b51f8ad740414d126c47b843 +size 542847136