fercho-05 commited on
Commit
07ab8e5
·
1 Parent(s): c652fe9
Files changed (1) hide show
  1. orderIA.py +4 -3
orderIA.py CHANGED
@@ -2,9 +2,7 @@ from transformers import AutoTokenizer, AutoModelForSequenceClassification, Trai
2
  import pandas as pd
3
  from datasets import Dataset
4
  from transformers import BertTokenizerFast
5
- # Carga del tokenizador
6
 
7
- #tokenizer = BertTokenizerFast.from_pretrained('Sebastian2903/SMARTORDERIA')
8
  # Cargar el dataset
9
  data = [
10
  {"text": "¿Qué hamburguesas tienen?", "label": 0},
@@ -17,7 +15,10 @@ df = pd.DataFrame(data)
17
  dataset = Dataset.from_pandas(df)
18
 
19
  # Tokenizar los datos
20
- tokenizer = AutoTokenizer.from_pretrained("bert-base-multilingual-cased")
 
 
 
21
 
22
  def preprocess_function(examples):
23
  return tokenizer(examples['text'], truncation=True, padding=True)
 
2
  import pandas as pd
3
  from datasets import Dataset
4
  from transformers import BertTokenizerFast
 
5
 
 
6
  # Cargar el dataset
7
  data = [
8
  {"text": "¿Qué hamburguesas tienen?", "label": 0},
 
15
  dataset = Dataset.from_pandas(df)
16
 
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):
24
  return tokenizer(examples['text'], truncation=True, padding=True)