File size: 2,214 Bytes
da58bf8 798274c da58bf8 798274c da58bf8 798274c |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 |
---
title: PaperClassification
emoji: 📚
colorFrom: gray
colorTo: yellow
models:
- oracat/bert-paper-classifier-arxiv
- oracat/bert-paper-classifier
sdk: streamlit
sdk_version: 1.44.1
app_file: app.py
pinned: false
short_description: Different models for paper classification
---
# Как запустить локально
Для начала нужно создать venv и загрузить зависимости (это нужно сделать только один раз). Для Linux-подобных систем это делается следующим образом:
```sh
python -m venv venv # Создать виртуальной среды
chmod +x venv/bin/activate # Добавить права на исполнение скрипту, который активирует среду
source venv/bin/activate # Активировать среду
pip install -r requirements.txt # Установить все зависимости
```
Находясь в корневой директории репозитория:
```sh
streamlit run app.py
```
Для выхода из виртуальной среды выполните:
```sh
deactivate
```
# Как добавить модель
В директории `category_classification/models` создайте директорию для новой модели. Название директории: либо `название_модели` либо `автор__название_модели`.
В этой директории создайте файл `model.py`, в котором опишите архитектуру новой модели. В этом же модуле обязательно должен быть строковый атрибут `name`, соответствующий имени новой модели и функция `get_model`, которая возвращает объект модели. Объект модели должен быть callable, в него будет передаваться аргумент типа `Input` (см. [common.py](src/common.py)).
Объект модели должен быть pickable.
|