clavel commited on
Commit
1694dfe
1 Parent(s): a388654

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -4
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. [Actuaría, Derecho, Economía, Psicología, Química Farmacéutico Biológica]"
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']