Naas drivers integration

Tags: #huggingface #nlp #huggingface #api #models #transformers #sales #ai #text
Author: Gagan Bhatia
In this notebook, you will be able to explore the Hugging Face transformers package with minimal technical knowledge thanks to Naas low-code drivers. Hugging Face is an immensely popular Python library providing pretrained models that are extraordinarily useful for a variety of natural language processing (NLP) tasks.

How it works?

Naas drivers HuggingFace formulas follow this format.
huggingface.get(task, model, tokenizer)(inputs)
The supported tasks are the following:
  • text-generation (model: GPT2)
  • summarization (model: t5-small)
  • fill-mask (model: distilroberta-base)
  • text-classification (model: distilbert-base-uncased-finetuned-sst-2-english)
  • feature-extraction (model: distilbert-base-cased)
  • token-classification (model: dslim/bert-base-NER)
  • question-answering
  • translation
We simply use Hugging Face API under the hood to access the models.


Import library

from naas_drivers import huggingface

Text Generation

huggingface.get("text-generation", model="gpt2", tokenizer="gpt2")("What is the most important thing in your life right now?")


Text Summarization

Summarize the text given, maximum lenght (number of tokens/words) is set to 200.
huggingface.get("summarization", model="t5-small", tokenizer="t5-small")('''
There will be fewer and fewer jobs that a robot cannot do better.
What to do about mass unemployment this is gonna be a massive social challenge and
I think ultimately we will have to have some kind of universal basic income.
I think some kind of a universal basic income is going to be necessary
now the output of goods and services will be extremely high
so with automation they will they will come abundance there will be or almost everything will get very cheap.
The harder challenge much harder challenge is how do people then have meaning like a lot of people
they find meaning from their employment so if you don't have if you're not needed if
there's not a need for your labor how do you what's the meaning if you have meaning
if you feel useless these are much that's a much harder problem to deal with.

Text Classification

Basic sentiment analysis on a text. Returns a "label" (negative/neutral/positive), and score between -1 and 1.
It was a weird concept. Why would I really need to generate a random paragraph?
Could I actually learn something from doing so?
All these questions were running through her head as she pressed the generate button.
To her surprise, she found what she least expected to see.

Fill Mask

Fill the blanks ('< mask >') in a sentence given with multiple proposals. Each proposal has a score (confidence of accuracy), token value (proposed word in number), token_str (proposed word)
It was a beautiful <mask>.

Feature extraction

This generate a words embedding (extract numbers out of the text data). Output is a list of numerical values.
huggingface.get("feature-extraction", model="distilbert-base-cased", tokenizer="distilbert-base-cased")("Life is a super cool thing")

Token classification

Basically NER. If you give names, location, or any "entity" it can detect it.
Entity abreviation
Outside of a named entity
Beginning of a miscellaneous entity right after another miscellaneous entity
Miscellaneous entity
Beginning of a person’s name right after another person’s name
Person’s name
Beginning of an organization right after another organization
Beginning of a location right after another location
Full documentation :


Display result

huggingface.get("token-classification", model="dslim/bert-base-NER", tokenizer="dslim/bert-base-NER")('''
My name is Wolfgang and I live in Berlin