Get startedGet started for free

Building a search completion system

Search completion, or auto-complete, is a common NLP application used in search engines and messaging apps. The goal is to suggest possible completions based on a user's partial input. Your task is to use Hugging Face's "text-generation" pipeline to implement a basic auto-complete system that generates relevant completions from the user's query.

This exercise is part of the course

Natural Language Processing (NLP) in Python

View Course

Exercise instructions

  • Create an autocomplete pipeline with the "distilgpt2" model.
  • Generate five search query suggestions for the given prompt, limiting each to a maximum of eight tokens.

Hands-on interactive exercise

Have a go at this exercise by completing this sample code.

from transformers import pipeline

# Create the pipeline
autocomplete = pipeline(task="____", model="____")

prompt = "Best books to read for"

# Generate search query completions
suggestions = ____

for suggestion in suggestions:
    print(suggestion['generated_text'])
Edit and Run Code