Instructions to use Master-AI-Lab/EnergyBERT with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use Master-AI-Lab/EnergyBERT with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("fill-mask", model="Master-AI-Lab/EnergyBERT")# Load model directly from transformers import AutoTokenizer, AutoModelForMaskedLM tokenizer = AutoTokenizer.from_pretrained("Master-AI-Lab/EnergyBERT") model = AutoModelForMaskedLM.from_pretrained("Master-AI-Lab/EnergyBERT", device_map="auto") - Notebooks
- Google Colab
- Kaggle
Update README.md
Browse files
README.md
CHANGED
|
@@ -2,7 +2,7 @@
|
|
| 2 |
license: mit
|
| 3 |
language:
|
| 4 |
- en
|
| 5 |
-
pipeline_tag:
|
| 6 |
---
|
| 7 |
|
| 8 |
# Model Details
|
|
@@ -23,5 +23,4 @@ Domain Specific BERT model for Text Mining in Energy & Material Field
|
|
| 23 |
<!-- Provide the basic links for the model. -->
|
| 24 |
|
| 25 |
- **Repository:** [Github](https://github.com/MasterAI-EAM/EnergyBERT)
|
| 26 |
-
- **Paper:**
|
| 27 |
-
|
|
|
|
| 2 |
license: mit
|
| 3 |
language:
|
| 4 |
- en
|
| 5 |
+
pipeline_tag: fill-mask
|
| 6 |
---
|
| 7 |
|
| 8 |
# Model Details
|
|
|
|
| 23 |
<!-- Provide the basic links for the model. -->
|
| 24 |
|
| 25 |
- **Repository:** [Github](https://github.com/MasterAI-EAM/EnergyBERT)
|
| 26 |
+
- **Paper:**
|
|
|