nero7151 commited on
Commit
08c0f22
·
1 Parent(s): 9fa053a

재업로드

Browse files
Files changed (2) hide show
  1. Dockerfile +1 -1
  2. app/main.py +0 -3
Dockerfile CHANGED
@@ -1,6 +1,6 @@
1
  FROM python:3.10
2
 
3
- ENV TRANSFORMERS_CACHE=/tmp/huggingface
4
 
5
  WORKDIR /app
6
 
 
1
  FROM python:3.10
2
 
3
+ ENV HF_HOME=/tmp/huggingface
4
 
5
  WORKDIR /app
6
 
app/main.py CHANGED
@@ -1,6 +1,3 @@
1
- import os
2
- os.environ["TRANSFORMERS_CACHE"] = "/tmp/huggingface"
3
-
4
  from fastapi import FastAPI, Request
5
  from transformers import pipeline
6
 
 
 
 
 
1
  from fastapi import FastAPI, Request
2
  from transformers import pipeline
3