Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -51,7 +51,8 @@ El presente conjunto de datos favorecerá la búsqueda e investigación relacion
|
|
51 |
tokenizer = AutoTokenizer.from_pretrained('hiiamsid/BETO_es_binary_classification', use_fast=False)
|
52 |
model = AutoModelForSequenceClassification.from_pretrained(
|
53 |
'hackathon-pln-es/unam_tesis_BETO_finnetuning', num_labels=5, output_attentions=False, output_hidden_states=False)
|
54 |
-
pipe = TextClassificationPipeline(model=model, tokenizer=tokenizer, return_all_scores=True)
|
|
|
55 |
|
56 |
#classificationResult = pipe("El objetivo de esta tesis es elaborar un estudio de las condiciones asociadas al aprendizaje desde casa.")
|
57 |
|
|
|
51 |
tokenizer = AutoTokenizer.from_pretrained('hiiamsid/BETO_es_binary_classification', use_fast=False)
|
52 |
model = AutoModelForSequenceClassification.from_pretrained(
|
53 |
'hackathon-pln-es/unam_tesis_BETO_finnetuning', num_labels=5, output_attentions=False, output_hidden_states=False)
|
54 |
+
#pipe = TextClassificationPipeline(model=model, tokenizer=tokenizer, return_all_scores=True)
|
55 |
+
pipe = pipeline (model='hackathon-pln-es/unam_tesis_BETO_finnetuning')
|
56 |
|
57 |
#classificationResult = pipe("El objetivo de esta tesis es elaborar un estudio de las condiciones asociadas al aprendizaje desde casa.")
|
58 |
|