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/ms-marco-MiniLM-L6-v2
- Downloads last month
- 4
Model tree for UNSW990025T2Transformer/Header_Extraction_MiniLM
Base model
microsoft/MiniLM-L12-H384-uncased Quantized
cross-encoder/ms-marco-MiniLM-L12-v2 Quantized
cross-encoder/ms-marco-MiniLM-L6-v2
# Load model directly from transformers import AutoTokenizer, AutoModelForSequenceClassification tokenizer = AutoTokenizer.from_pretrained("UNSW990025T2Transformer/Header_Extraction_MiniLM") model = AutoModelForSequenceClassification.from_pretrained("UNSW990025T2Transformer/Header_Extraction_MiniLM")