Skip to content

xbtart/PE_itog

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

16 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Итоговый проект - Веб-приложение для эмоционального анализа текста на основе RuBERT

Это веб-приложение использует FastAPI для создания REST API, которое позволяет анализировать эмоциональный тон текстов с использованием модели cointegrated/rubert-tiny2-cedr-emotion-detection.

Установка и запуск

  1. Установите библиотеки, упомянутые в файле requirements.txt, используя следующую команду:
pip install -r requirements.txt
  1. Запустите сервер с помощью команды:
uvicorn main:app --host 0.0.0.0 --port 8000

API endpoints

POST /predict

Позволяет передать текст для анализа эмоционального тона.

Параметры запроса

  • text (строка): Текст для анализа.

Пример запроса:

curl -X POST -H "Content-Type: application/json; charset=utf-8" -d "{\"text\": \"Я очень удивлен происходящим!\"}" http:https://xbondteam.ru:8000/predict

Пример успешного ответа:

{"текст":"Я очень удивлен происходящим!","эмоции":{"грусть":0.0012,"радость":0.0078,"злость":0.0012,"страх":0.9871,"удивление":0.0016}}

Возвращается список вероятностей для каждой из пяти эмоций: ["грусть", "радость", "злость", "страх", "удивление"].

Пример использования скрипта send_request.py

import requests

text = "Я очень удивлен происходящим!"
url = "http:https://127.0.0.1:8000/predict"

payload = {"text": text}
response = requests.post(url, json=payload)

print(response.json())

Зависимости

  • fastapi
  • transformers
  • torch
  • pydantic
  • decimal
  • requests

Запуск тестов

Чтобы запустить тесты, выполните команду:

pytest

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages