Skip to content

ball2004244/LogosDB-AI-Models

Repository files navigation

LogosDB-AI-Models

This repository contains the code for the AI models used in the LogosDB project.

  1. Extractive Summary: extract portions of sentences that have the highest importance in the text using Tf-idf and TextRank algorithms.

  2. Abstractive Summary: finetune Google T5 model on CNN/DailyMail Dataset.

Installation

  1. Clone the repository
  2. Install the required packages (Different methods have different requirements, see below) 2.1 Keyword Summary:
pip install keybert

2.2 Extractive Summary: For extractive summary requirements, install the following:

pip install cython nltk networkx numpy scikit-learn

Compile the cython code:

cd extractive_sum/cython
python3 setup.py build_ext --inplace

To use compiled Cython code, import the compiled module to your C++ code:

#include "extractive_sum/cython/summarizer.h"

2.3 Abstractive Summary: Install requirements:

pip install transformers torch peft datasets pandas

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

 
 
 

Contributors