Spaces:
Running
Running
File size: 421 Bytes
00f9b6a |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 |
# from transformers import pipeline
# from summarizer import Summarizer
# categorizing_model = pipeline("zero-shot-classification", model='cross-encoder/nli-deberta-v3-base')
# while True:
# print('Enter text:')
# to_cat = input()
# result = categorizing_model(
# to_cat,
# candidate_labels=Summarizer.ARTICLE_CATEGORIES
# )
# print('\n\nResult:')
# print(result['labels'][0]) |