Hugging Face Summarization

Summarization using models hosted in Hugging Face repository.

Input

It takes in a DataFrame as input

Type

ml-estimator

Class

fire.nodes.gai.NodeHFSummarization

Fields

Name

Title

Description

column_name

Column to summarize

Column to summarize.

preloaded_model

Model to be used for Summarization

Summarizes the content using the model chosen.

preloaded_model_path

Path

Pre-loaded models parent directory.

custom_model

Custom model name

Model name to pull model from Hugging Face repo.