islasher commited on
Commit
a80504a
1 Parent(s): dc25c60

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -8,14 +8,14 @@ import torch
8
  # repo_id = "YOUR_USERNAME/YOUR_LEARNER_NAME"
9
  repo_id = "islasher/clasificador-dair-emotion"
10
 
11
- learner = from_pretrained_fastai(repo_id)
12
- labels = learner.dls.vocab
13
 
14
  # Definimos una función que se encarga de llevar a cabo las predicciones
15
 
16
 
17
  # Cargar el modelo y el tokenizador
18
- nombre_modelo = "clasificador-dair-emotion"
19
  model = AutoModelForSequenceClassification.from_pretrained(nombre_modelo)
20
  tokenizer = AutoTokenizer.from_pretrained(nombre_modelo)
21
 
 
8
  # repo_id = "YOUR_USERNAME/YOUR_LEARNER_NAME"
9
  repo_id = "islasher/clasificador-dair-emotion"
10
 
11
+ # learner = from_pretrained_fastai(repo_id)
12
+ # labels = learner.dls.vocab
13
 
14
  # Definimos una función que se encarga de llevar a cabo las predicciones
15
 
16
 
17
  # Cargar el modelo y el tokenizador
18
+ nombre_modelo = "islasher/clasificador-dair-emotion"
19
  model = AutoModelForSequenceClassification.from_pretrained(nombre_modelo)
20
  tokenizer = AutoTokenizer.from_pretrained(nombre_modelo)
21