fercho-05 commited on
Commit
eb04d05
·
1 Parent(s): 07ab8e5
Files changed (1) hide show
  1. orderIA.py +2 -2
orderIA.py CHANGED
@@ -17,7 +17,7 @@ dataset = Dataset.from_pandas(df)
17
  # Tokenizar los datos
18
  # Carga del tokenizador
19
 
20
- tokenizer = BertTokenizerFast.from_pretrained("bert-base-multilingual-cased")
21
  #tokenizer = AutoTokenizer.from_pretrained("bert-base-multilingual-cased")
22
 
23
  def preprocess_function(examples):
@@ -31,7 +31,7 @@ train_dataset = train_test_split['train']
31
  eval_dataset = train_test_split['test']
32
 
33
  # Configurar el modelo
34
- model = AutoModelForSequenceClassification.from_pretrained("bert-base-multilingual-cased", num_labels=4)
35
 
36
  # Configurar el entrenador
37
  training_args = TrainingArguments(
 
17
  # Tokenizar los datos
18
  # Carga del tokenizador
19
 
20
+ tokenizer = BertTokenizerFast.from_pretrained("Sebastian2903/SMARTORDERIA")
21
  #tokenizer = AutoTokenizer.from_pretrained("bert-base-multilingual-cased")
22
 
23
  def preprocess_function(examples):
 
31
  eval_dataset = train_test_split['test']
32
 
33
  # Configurar el modelo
34
+ model = AutoModelForSequenceClassification.from_pretrained("Sebastian2903/SMARTORDERIA", num_labels=4)
35
 
36
  # Configurar el entrenador
37
  training_args = TrainingArguments(