somosnlp-hackathon-2022
/

unam_tesis_BETO_finnetuning

Text Classification

Inference Endpoints

Model card Files Files and versions Community

clavel commited on Apr 13, 2022

Commit

e0c7c5e

•

1 Parent(s): 75532ac

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -44,7 +44,7 @@ For further details on how to use unam_tesis_BETO_finnetuning you can visit the
  tokenizer = AutoTokenizer.from_pretrained('hiiamsid/BETO_es_binary_classification', use_fast=False)
  model = AutoModelForSequenceClassification.from_pretrained(
-                   'hackathon-pln-e/unam_tesis_BETO_finnetuning', num_labels=5, output_attentions=False,
                   output_hidden_states=False)
  pipe = TextClassificationPipeline(model=model, tokenizer=tokenizer, return_all_scores=True)

  tokenizer = AutoTokenizer.from_pretrained('hiiamsid/BETO_es_binary_classification', use_fast=False)
  model = AutoModelForSequenceClassification.from_pretrained(
+                   'hackathon-pln-es/unam_tesis_BETO_finnetuning', num_labels=5, output_attentions=False,
                   output_hidden_states=False)
  pipe = TextClassificationPipeline(model=model, tokenizer=tokenizer, return_all_scores=True)