Marroco93 commited on
Commit
04ac801
1 Parent(s): 7bc74bc

no message

Browse files
Files changed (2) hide show
  1. main.py +1 -0
  2. requirements.txt +1 -0
main.py CHANGED
@@ -8,6 +8,7 @@ from typing import Generator
8
  import json # Asegúrate de que esta línea esté al principio del archivo
9
  import nltk
10
  import os
 
11
  from transformers import pipeline, AutoTokenizer,AutoModelForSeq2SeqLM
12
 
13
 
 
8
  import json # Asegúrate de que esta línea esté al principio del archivo
9
  import nltk
10
  import os
11
+ import sentencepiece
12
  from transformers import pipeline, AutoTokenizer,AutoModelForSeq2SeqLM
13
 
14
 
requirements.txt CHANGED
@@ -5,3 +5,4 @@ pydantic
5
  torch==2.0.0
6
  nltk
7
  transformers
 
 
5
  torch==2.0.0
6
  nltk
7
  transformers
8
+ sentencepiece