File size: 2,278 Bytes
d0fdaa7 61fabaf 4dcef19 d0fdaa7 0e2fe46 1ebb005 0e2fe46 d0fdaa7 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 |
---
datasets:
- sms_spam
pipeline_tag: text-classification
license: mit
language:
- en
library_name: fasttext, bert
---
## What is Open Source Text Shield (OTS)?
OTS (Open Source Text Shield) is an AI-driven solution designed to enhance the security of telecom networks by detecting and filtering spam and phishing messages in real time. This application leverages both BERT and FastText models for efficient text classification.
## Getting Started
### Prerequisites
- Python 3.8 or later
- FastAPI
- pydantic
- torch
- transformers
- fasttext
- BERT
You can install the necessary libraries using pip:
```bash
pip install fastapi pydantic torch transformers fasttext
```
### Installation
Clone the repository to your local machine:
```bash
git clone https://github.com/TelecomsXChangeAPi/OpenTextShield/
```
Navigate to the cloned directory:
```bash
cd OpenTextShield
```
### Running the Application
Start the server by running:
```bash
uvicorn main:app --host 0.0.0.0 --port 8001
```
The application will be available at `http://localhost:8001`.
### Usage
#### Predicting SMS
To predict if an SMS is spam, phishing, or ham (regular message), send a POST request to `/predict/` with a JSON body containing the SMS text and the model to use (`bert` or `fasttext`).
Example using curl:
```bash
curl -X POST "http://localhost:8001/predict/" -H "accept: application/json" -H "Content-Type: application/json" -d "{\"text\":\"Your SMS content here\",\"model\":\"bert\"}"
```
#### Feedback Loop
To provide feedback on predictions, send a POST request to `/feedback-loop/` with relevant feedback data.
Example using curl:
```bash
curl -X POST "http://localhost:8001/feedback-loop/" -H "accept: application/json" -H "Content-Type: application/json" -d "{\"content\":\"SMS content\",\"feedback\":\"Your feedback here\",\"thumbs_up\":true,\"thumbs_down\":false,\"user_id\":\"user123\",\"model\":\"bert\"}"
```
#### Download Feedback
To download the feedback data for a specific model, send a GET request to `/download-feedback/{model_name}`.
Example using curl:
```bash
curl -X GET "http://localhost:8001/download-feedback/bert"
```
## Acknowledgements
Special thanks to the team at TelecomsXChange (TCXC) for their invaluable contributions to this project. |