Créer un système de complétion de recherche
La complétion de recherche, ou l’auto-complétion, est une application courante du NLP utilisée dans les moteurs de recherche et les applications de messagerie. L’objectif est de suggérer des complétions possibles à partir de la saisie partielle d’un utilisateur. Votre tâche consiste à utiliser le pipeline "text-generation" de Hugging Face pour implémenter un système d’auto-complétion basique qui génère des complétions pertinentes à partir de la requête de l’utilisateur.
Cet exercice fait partie du cours
Natural Language Processing (NLP) in Python
Instructions
- Créez un pipeline d’
autocompleteavec le modèle"distilgpt2". - Générez cinq
suggestionsde requêtes de recherche pour l’promptdonné, en limitant chacune à un maximum de huit jetons.
Exercice interactif pratique
Essayez cet exercice en complétant cet exemple de 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'])