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
Exercise instructions
- Create an
autocomplete
pipeline with the"distilgpt2"
model. - Generate five search query
suggestions
for the givenprompt
, 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'])