3v324v23 commited on
Commit
a2997f8
1 Parent(s): d8a2329
Files changed (1) hide show
  1. Dockerfile +2 -0
Dockerfile CHANGED
@@ -38,6 +38,8 @@ RUN pip install --no-cache-dir --upgrade -r requirements.txt
38
 
39
  RUN wget https://www.dropbox.com/scl/fi/105qy7mkqfjcmnfd3tmv0/edit.pth?rlkey=qcd67cdrqz4jra0p3er966iuk -O clevr.pth
40
 
 
 
41
 
42
  # CMD ["uvicorn", "app.main:app", "--host", "0.0.0.0", "--port", "7860"]
43
  CMD ["python", "app.py"]
 
38
 
39
  RUN wget https://www.dropbox.com/scl/fi/105qy7mkqfjcmnfd3tmv0/edit.pth?rlkey=qcd67cdrqz4jra0p3er966iuk -O clevr.pth
40
 
41
+ ENV TORCH_EXTENSIONS_DIR=/home/user/.cache
42
+
43
 
44
  # CMD ["uvicorn", "app.main:app", "--host", "0.0.0.0", "--port", "7860"]
45
  CMD ["python", "app.py"]