Update README.md
Browse files
README.md
CHANGED
@@ -44,7 +44,7 @@ For further details on how to use unam_tesis_BETO_finnetuning you can visit the
|
|
44 |
|
45 |
tokenizer = AutoTokenizer.from_pretrained('hiiamsid/BETO_es_binary_classification', use_fast=False)
|
46 |
model = AutoModelForSequenceClassification.from_pretrained(
|
47 |
-
'hackathon-pln-
|
48 |
output_hidden_states=False)
|
49 |
pipe = TextClassificationPipeline(model=model, tokenizer=tokenizer, return_all_scores=True)
|
50 |
|
|
|
44 |
|
45 |
tokenizer = AutoTokenizer.from_pretrained('hiiamsid/BETO_es_binary_classification', use_fast=False)
|
46 |
model = AutoModelForSequenceClassification.from_pretrained(
|
47 |
+
'hackathon-pln-es/unam_tesis_BETO_finnetuning', num_labels=5, output_attentions=False,
|
48 |
output_hidden_states=False)
|
49 |
pipe = TextClassificationPipeline(model=model, tokenizer=tokenizer, return_all_scores=True)
|
50 |
|