idiom-xlm-roberta / README.md
imranraad's picture
Update README.md
ba4deea
metadata
tags:
  - autotrain
  - token-classification
language:
  - en
widget:
  - text: >-
      He at the last appointed him on one, And let all others from his hearte
      gon, And chose her of his own authority; For love is blind all day, and
      may not see.
  - text: >-
      I’m sorry but I just can’t seem to wrap my head around it. - I’m sorry but
      I just can’t seem to understand.
  - text: Why are you so bent out of shape? - Why are you so upset?
  - text: Listen, it is easier said than done, many people lack commitment.
co2_eq_emissions:
  emissions: 0.04215761331893144
license: mit
library_name: transformers
pipeline_tag: token-classification

Fine-tune datasets

Model Trained Using AutoTrain

  • Problem type: Entity Extraction
  • Model ID: 1595156286
  • CO2 Emissions (in grams): 0.0422

Validation Metrics

  • Loss: 0.012
  • Accuracy: 0.996
  • Precision: 0.000
  • Recall: 0.000
  • F1: 0.000

Usage

You can use cURL to access this model:

$ curl -X POST -H "Authorization: Bearer YOUR_API_KEY" -H "Content-Type: application/json" -d '{"inputs": "I love AutoTrain"}' https://api-inference.huggingface.co/models/imranraad/autotrain-magpie-epie-combine-xlmr-metaphor-1595156286

Or Python API:

from transformers import AutoModelForTokenClassification, AutoTokenizer

model = AutoModelForTokenClassification.from_pretrained("imranraad/autotrain-magpie-epie-combine-xlmr-metaphor-1595156286", use_auth_token=True)

tokenizer = AutoTokenizer.from_pretrained("imranraad/autotrain-magpie-epie-combine-xlmr-metaphor-1595156286", use_auth_token=True)

inputs = tokenizer("I love AutoTrain", return_tensors="pt")

outputs = model(**inputs)

How to get the idioms:

from transformers import AutoTokenizer, AutoModelForTokenClassification
from transformers import pipeline

model = AutoModelForTokenClassification.from_pretrained("imranraad/idiom-xlm-roberta")

tokenizer = AutoTokenizer.from_pretrained("imranraad/idiom-xlm-roberta")

pipeline_idioms = pipeline("ner", model=model, tokenizer=tokenizer, aggregation_strategy="simple")

text = "Why are you so bent out of shape? - Why are you so upset?"

idioms = pipeline_idioms(text)
for idiom in idioms:
    if idiom['entity_group'] == '1':
        print(idiom['word'])