25 lines
845 B
Python
25 lines
845 B
Python
|
#!/usr/bin/env python3
|
|||
|
|
import sys
|
|||
|
|
import json
|
|||
|
|
from deeppavlov import build_model, configs
|
|||
|
|
|
|||
|
|
def classify_dispute(text):
|
|||
|
|
# Загружаем базовую модель классификации из DeepPavlov
|
|||
|
|
classifier = build_model(configs.classifiers.torch_classification_model, download=True)
|
|||
|
|
result = classifier([text])
|
|||
|
|
# Предполагается, что результат — это список, содержащий метку класса
|
|||
|
|
return result[0]
|
|||
|
|
|
|||
|
|
def main():
|
|||
|
|
if len(sys.argv) < 2:
|
|||
|
|
print("Usage: classify_dispute.py \"<текст>\"")
|
|||
|
|
sys.exit(1)
|
|||
|
|
|
|||
|
|
input_text = " ".join(sys.argv[1:])
|
|||
|
|
essence = classify_dispute(input_text)
|
|||
|
|
output = {"суть_спора": essence}
|
|||
|
|
print(json.dumps(output, ensure_ascii=False, indent=2))
|
|||
|
|
|
|||
|
|
if __name__ == '__main__':
|
|||
|
|
main()
|