Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -3,7 +3,7 @@ from transformers import AutoTokenizer, AutoModelForSequenceClassification, Text
|
|
3 |
|
4 |
title = "Clasificador de Tesis"
|
5 |
|
6 |
-
description = "Clasifica el texto de una tesis con la carrera a la que le pertenece
|
7 |
|
8 |
article = """
|
9 |
## Obtención de los datos
|
@@ -57,9 +57,6 @@ pipe = TextClassificationPipeline(model=model, tokenizer=tokenizer, return_all_s
|
|
57 |
|
58 |
|
59 |
def thesis_prediction(input):
|
60 |
-
tokenizer = AutoTokenizer.from_pretrained('"hiiamsid/BETO_es_binary_classification"', use_fast=False)
|
61 |
-
model = AutoModelForSequenceClassification.from_pretrained('hackathon-pln-es/unam_tesis_BETO_finnetuning', num_labels=5, output_attentions=False, output_hidden_states=False)
|
62 |
-
pipe = TextClassificationPipeline(model=model, tokenizer=tokenizer, return_all_scores=True)
|
63 |
classificationResult = pipe(input)
|
64 |
classificationResult[0].sort(reverse=True, key=lambda x:x['score'])
|
65 |
keyClass = classificationResult[0][0]['label']
|
|
|
3 |
|
4 |
title = "Clasificador de Tesis"
|
5 |
|
6 |
+
description = "Clasifica el texto de una tesis con la carrera a la que le pertenece [Actuaría, Derecho, Economía, Psicología, Química Farmacéutico Biológica]."
|
7 |
|
8 |
article = """
|
9 |
## Obtención de los datos
|
|
|
57 |
|
58 |
|
59 |
def thesis_prediction(input):
|
|
|
|
|
|
|
60 |
classificationResult = pipe(input)
|
61 |
classificationResult[0].sort(reverse=True, key=lambda x:x['score'])
|
62 |
keyClass = classificationResult[0][0]['label']
|