khaiphan29 commited on
Commit
e265cce
1 Parent(s): ed22e94

Upload folder using huggingface_hub

Browse files
Files changed (2) hide show
  1. Dockerfile +1 -0
  2. src/myNLI.py +1 -1
Dockerfile CHANGED
@@ -11,4 +11,5 @@ RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
11
 
12
  COPY . .
13
 
 
14
  CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "7860"]
 
11
 
12
  COPY . .
13
 
14
+ RUN ["python3", "-m", "nltk.downloader", "all"]
15
  CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "7860"]
src/myNLI.py CHANGED
@@ -18,7 +18,7 @@ class FactChecker:
18
 
19
  @timer_func
20
  def __init__(self):
21
- nltk.download('punkt')
22
  self.INPUT_TYPE = "mean"
23
  self.load_model()
24
 
 
18
 
19
  @timer_func
20
  def __init__(self):
21
+ # nltk.download('punkt')
22
  self.INPUT_TYPE = "mean"
23
  self.load_model()
24