Opennyai is a python library for natural language preprocessing on Indian legal texts.
This library provides unified access to the following 3 pre-trained AI models developed by OpenNyAI which focus on Indian court judgments:
- Named Entity Recognition (NER): GitHub , paper, Huggingface Datasets
- Judgment Structuring using Sentence Rhetorical Roles: GitHub , paper, Huggingface Datasets
- Extractive Summarizer: GitHub
This library is mainly for running the pretrained models on your custom input judgments text. For more details about data and model training, please refer to individual git repo links.
To get started using opennyai first create a new python virtual environment using conda:
Supports python 3.8, 3.9, 3.10
conda create -n opennyai python=3.8
conda activate opennyaiInstall it using pip by running the following line in your terminal
pip install -U opennyaiIf you want to utilize spacy with GPU please install Cupy and cudatoolkit dependency of appropriate version.
conda install cudatoolkit==<your_cuda_version> #### E.g. cudatoolkit==11.2
pip install cupy-cuda<your_cuda_version> ##### E.g. cupy-cuda112In case of any issue with installation please refer to spacy installation with cupy
Remember you need spacy of 3.2.4 version for models to work perfectly.
Please refer to the Documentation for more details.
To run the 3 OpenNyAI models on judgment texts of your choice please run following python code
from opennyai import Pipeline
from opennyai.utils import Data
import urllib
# Get court judgment texts on which to run the AI models
text1 = urllib.request.urlopen(
'https://raw.githubusercontent.com/OpenNyAI/Opennyai/master/samples/sample_judgment1.txt').read().decode()
text2 = urllib.request.urlopen(
'https://raw.githubusercontent.com/OpenNyAI/Opennyai/master/samples/sample_judgment2.txt').read().decode()
# you can also load your text files directly into this
texts_to_process = [text1, text2]
# create Data object for data preprocessing before running ML models
data = Data(texts_to_process)
# If you have access to GPU then set this to True else False
use_gpu = True
# Choose which of the AI models you want to run from the 3 models 'NER', 'Rhetorical_Role','Summarizer'. E.g. If just Named Entity is of interest then just select 'NER'
pipeline = Pipeline(components=['NER', 'Rhetorical_Role', 'Summarizer'], use_gpu=use_gpu, verbose=True)
results = pipeline(data)The output of each model is present in following keys of each element of the output
results[0][
'annotations'] ## shows the result of model at sentence level, each entry will have entities, rhetorical role, and other details
results[0]['summary'] ## shows Summary for each of the Rheorical Role for first judgment text For more details on usage please refer to the documentation
We encourage you to use the local machine installation of opennyai library instead of Google colab as it works with python 3.8. Since Google colab works with higher versions of python, opennyai installation does not work.
| Description | Link |
|---|---|
| Run Inference |  |
We encourage users to use our webapp for visualizing the results for a judgment of your choice.