How to use from the
Use from the
Transformers library
# Load model directly
from transformers import AutoTokenizer, AutoModelForSequenceClassification

tokenizer = AutoTokenizer.from_pretrained("UNSW990025T2Transformer/Header_Extraction_Robert")
model = AutoModelForSequenceClassification.from_pretrained("UNSW990025T2Transformer/Header_Extraction_Robert")
Quick Links

Model Card for Model ID

Match the most appropriate header from the table based on the target field and the content of the corresponding column.

Model Details

Model Description

  • Developed by: UNSW COMP9900 25T2 BREAD Transformer Team
  • Model type: Transformer
  • Language (NLP): English
  • License: MIT
  • Finetuned from model : cross-encoder/stsb-roberta-base
Downloads last month
3
Safetensors
Model size
0.1B params
Tensor type
F32
·
Inference Providers NEW
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support

Model tree for UNSW990025T2Transformer/Header_Extraction_Robert

Finetuned
(6)
this model