Створення системи автодоповнення пошуку
Автодоповнення пошуку — поширений застосунок NLP у пошукових системах і месенджерах. Мета — запропонувати можливі варіанти завершення на основі часткового введення користувача. Ваше завдання — використати pipeline Hugging Face "text-generation", щоб реалізувати базову систему автодоповнення, яка генерує доречні завершення з користувацького запиту.
Ця вправа є частиною курсу
Natural Language Processing (NLP) in Python
Інструкції до вправи
- Створіть pipeline
autocompleteз моделлю"distilgpt2". - Згенеруйте пʼять пошукових запитів
suggestionsдля заданогоprompt, обмеживши кожен щонайбільше вісьмома токенами.
Інтерактивна практична вправа
Спробуйте виконати цю вправу, доповнивши цей зразок коду.
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'])