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])