clavel commited on
Commit
ac37333
1 Parent(s): 97626a9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -52,7 +52,7 @@ tokenizer = AutoTokenizer.from_pretrained('hiiamsid/BETO_es_binary_classificatio
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', tokenizer=tokenizer, return_all_scores=True)
56
  #classificationResult = pipe("El objetivo de esta tesis es elaborar un estudio de las condiciones asociadas al aprendizaje desde casa.")
57
 
58
 
 
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', tokenizer=tokenizer)
56
  #classificationResult = pipe("El objetivo de esta tesis es elaborar un estudio de las condiciones asociadas al aprendizaje desde casa.")
57
 
58