Word from creator
Hello👋, This package is supports using Tensorflow Lite models for text classifications. I developed this when I wanted to integrate some models I generated from Tensorflow Model Maker in my flutter app.
Yes, without a doubt, giving a free 👍 or ⭐ will encourage me to keep this plugin updated.
Package description
A flutter plugin for doing text classification using tflite models.
Note: This project utilises tensorflow-lite-task-text library to classify the texts through models.
Features
- Works on Android 5.0 (API level 21) or later.
- Requires 4 line code to get running.
Getting started
- In pubspec.yaml, add this dependency:
tflite_text_classification:
- Add this package to your project:
import 'package:tflite_text_classification/tflite_text_classification.dart';
Basic Usage
ClassificationResult? result = await TfliteTextClassification().classifyText(
params: TextClassifierParams params = TextClassifierParams(
text: 'aaj me bahut khush hu',
modelPath:
'path/mobilebert.tflite',
modelType: ModelType.mobileBert,
delegate: 0,
),
);