Hugging Face's logo

Sebastian2903
/
SMARTORDERIA

Audio-to-Audio
Transformers
Spanish
bert
text-classification
Model card Files
xet
 Community
fercho-05 commited on
Commit
8ba4893
·
1 Parent(s): 4ed2e9b

Cambios 2

Browse files
Files changed (1) hide show
  1. orderIA.py +1 -1
orderIA.py CHANGED
@@ -30,7 +30,7 @@ train_dataset = train_test_split['train']
30
  eval_dataset = train_test_split['test']
31
 
32
  # Configurar el modelo
33
- model = AutoModelForSequenceClassification.from_pretrained("bert-base-multilingual-cased", num_labels=4)
34
 
35
  # Configurar el entrenador
36
  training_args = TrainingArguments(
 
30
  eval_dataset = train_test_split['test']
31
 
32
  # Configurar el modelo
33
+ model = AutoModelForSequenceClassification.from_pretrained('Sebastian2903/SMARTORDERIA', num_labels=4)
34
 
35
  # Configurar el entrenador
36
  training_args = TrainingArguments(