Spaces:
Runtime error
Runtime error
title: Project1 | |
emoji: π’ | |
colorFrom: yellow | |
colorTo: red | |
sdk: streamlit | |
sdk_version: 1.31.1 | |
app_file: app.py | |
pinned: false | |
license: apache-2.0 | |
# Streamlit app for NLP tasks π‘ | |
1. Film review classification on 'Good', 'Bad', 'Neutral' classes | |
2. Assessing the toxicity of a user message | |
3. Text generation using the GPT model using a custom prompt | |
## Team π§π»βπ» | |
1. [Alexey Kamaev](https://github.com/AlexeyKamaev) | |
2. [Marina Kochetova](https://github.com/neonanet) | |
3. [Valeriia Dashieva](https://github.com/valeriedaash) | |
## Used models π€ | |
1. For classification task we trained 3 models: TF-IDF vectorizer + LogReg, Word2Vec + LogReg, RuBert + LogReg | |
2. For toxicity assessing task we trained rubert-tiny-toxicity | |
3. For Text generation task we trained rugpt3small_based_on_gpt2 | |
## Used instruments π§° | |
1. Python. | |
2. Pytorch. | |
3. Transformers | |
4. [Streamlit](https://nnproject1-6tcyg5tnwo2we6fg8gqbgt.streamlit.app). | |