hghcomphys's picture
Update README.md
ae209e0 verified
|
raw
history blame
No virus
888 Bytes
metadata
language:
  - nl
thumbnail: >-
  https://raw.githubusercontent.com/hghcomphys/hghcomphys.github.io/master/assets/urls/GEOBERTJE.png
tags:
  - GEOBERTje
A domain-adapted language model trained on geological borehole descriptions in the Dutch language from Flanders region in Belgium.

How to use GEOBERTje as a pipeline:

from transformers import pipeline

pipe = pipeline("fill-mask", model="hghcomphys/geobertje-base-dutch-uncased")
pipe("grijs fijn zand met enkele [MASK]")

Load model directly:

from transformers import AutoTokenizer, AutoModel

tokenizer = AutoTokenizer.from_pretrained("hghcomphys/geobertje-base-dutch-uncased")
model = AutoModel.from_pretrained("hghcomphys/geobertje-base-dutch-uncased")