Crear un sistema de autocompletado de búsquedas
El autocompletado es una aplicación habitual de NLP en motores de búsqueda y apps de mensajería. Su objetivo es sugerir posibles continuaciones a partir de una entrada parcial del usuario. Tu tarea es usar el pipeline de Hugging Face "text-generation" para implementar un autocompletado básico que genere completions relevantes a partir de la consulta del usuario.
Este ejercicio forma parte del curso
Natural Language Processing (NLP) en Python
Instrucciones del ejercicio
- Crea un pipeline
autocompletecon el modelo"distilgpt2". - Genera cinco
suggestionsde consulta de búsqueda para elpromptdado, limitando cada una a un máximo de ocho tokens.
Ejercicio interactivo práctico
Prueba este ejercicio y completa el código de muestra.
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'])