README.md

metadata

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-подобных систем это делается следующим образом:

    python -m venv venv               # Создать виртуальной среды
    chmod +x venv/bin/activate        # Добавить права на исполнение скрипту, который активирует среду
    source venv/bin/activate          # Активировать среду
    pip install -r requirements.txt   # Установить все зависимости

Находясь в корневой директории репозитория:

    streamlit run app.py

Для выхода из виртуальной среды выполните:

    deactivate

Как добавить модель

В директории category_classification/models создайте директорию для новой модели. Название директории: либо название_модели либо автор__название_модели.
В этой директории создайте файл model.py, в котором опишите архитектуру новой модели. В этом же модуле обязательно должен быть строковый атрибут name, соответствующий имени новой модели и функция get_model, которая возвращает объект модели. Объект модели должен быть callable, в него будет передаваться аргумент типа Input (см. common.py).
Объект модели должен быть pickable.