CoolFace
Apppublic

albertmartinez/openalex-topic-classification

sourceHugging Facemitupdated 1y agoView on Hugging Face
1likes
App README

OpenAlex Topic Classification

This application allows you to classify academic texts into different topics using machine learning models trained with OpenAlex data.

Features

  • —Classification of academic texts into multiple topics
  • —Uses two different models for more robust classification
  • —Easy-to-use web interface
  • —Support for structured title and abstract format

Requirements

  • —Python 3.7+
  • —Gradio 5.23.1
  • —Transformers (Hugging Face)

Installation

bash
pip install -r requirements.txt

Usage

  1. 1.Run the application:
bash
python app.py
  1. 1.Open your browser at the address shown in the console (usually http://localhost:7860)
  1. 1.Enter your text in the format:
<TITLE> Your title here
<ABSTRACT> Your abstract here
  1. 1.Select the number of classifications you want to see (top_k)
  1. 1.Click "Submit" to get the results

Models

The application uses two different models:

  1. 1.OpenAlex/bert-base-multilingual-cased-finetuned-openalex-topic-classification-title-abstract
  2. 2.Based on BERT multilingual model
  3. 3.Fine-tuned on OpenAlex data
  4. 4.Supports multiple languages
  1. 1.albertmartinez/openalex-topic-classification-title-abstract
  2. 2.Based on BERT multilingual model
  3. 3.Fine-tuned on OpenAlex data (https://huggingface.co/datasets/albertmartinez/openalex-topic-title-abstract)
  4. 4.Supports multiple languages

License

MIT

References