clavel commited on
Commit
95c7e28
1 Parent(s): 3f40381

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -49,10 +49,11 @@ El presente conjunto de datos favorecerá la búsqueda e investigación relacion
49
  """
50
 
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("text-classification", model='hackathon-pln-es/unam_tesis_BETO_finnetuning')
56
  #classificationResult = pipe("El objetivo de esta tesis es elaborar un estudio de las condiciones asociadas al aprendizaje desde casa.")
57
 
58
 
 
49
  """
50
 
51
  #tokenizer = AutoTokenizer.from_pretrained('hiiamsid/BETO_es_binary_classification', use_fast=False)
52
+ tokenizer = AutoTokenizer.from_pretrained('GavinQiangLi/my-new-BertTokenizerFast-tokenizer', use_fast=False)
53
  #model = AutoModelForSequenceClassification.from_pretrained(
54
  # 'hackathon-pln-es/unam_tesis_BETO_finnetuning', num_labels=5, output_attentions=False, output_hidden_states=False)
55
  #pipe = TextClassificationPipeline(model=model, tokenizer=tokenizer, return_all_scores=True)
56
+ pipe = pipeline("text-classification", model='hackathon-pln-es/unam_tesis_BETO_finnetuning', tokenizer=tokenizer)
57
  #classificationResult = pipe("El objetivo de esta tesis es elaborar un estudio de las condiciones asociadas al aprendizaje desde casa.")
58
 
59