dhruv107 commited on
Commit
ffaaaad
·
1 Parent(s): ca25bd0
Files changed (2) hide show
  1. Dockerfile +2 -0
  2. pretrained.py +0 -3
Dockerfile CHANGED
@@ -6,6 +6,8 @@ WORKDIR /app
6
 
7
  ADD . /app
8
 
 
 
9
  # RUN apt-get update && apt-get install ffmpeg libsm6 libxext6 -y
10
 
11
  # RUN apt update
 
6
 
7
  ADD . /app
8
 
9
+ RUN export TRANSFORMERS_CACHE=/app
10
+
11
  # RUN apt-get update && apt-get install ffmpeg libsm6 libxext6 -y
12
 
13
  # RUN apt update
pretrained.py CHANGED
@@ -8,9 +8,6 @@ from flask import Flask, request, jsonify
8
  from werkzeug.utils import secure_filename
9
  # import logging
10
 
11
- import os
12
- os.environ['TRANSFORMERS_CACHE'] = '/app/API_LOGS'
13
-
14
 
15
  # # Set up root logger, and add a file handler to root logger
16
  # logging.basicConfig(filename = 'log_file.log',
 
8
  from werkzeug.utils import secure_filename
9
  # import logging
10
 
 
 
 
11
 
12
  # # Set up root logger, and add a file handler to root logger
13
  # logging.basicConfig(filename = 'log_file.log',