focal / app /test.py
michaelkri
Revamped front end design
00f9b6a
# 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])