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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
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