programindz commited on
Commit
c77ada1
·
1 Parent(s): f2b7d66

New Model Added

Browse files
Files changed (2) hide show
  1. Dockerfile +5 -1
  2. app/main.py +0 -1
Dockerfile CHANGED
@@ -9,9 +9,13 @@ RUN apt-get update && apt-get install -y \
9
  RUN git clone https://github.com/VarunGumma/IndicTransToolkit \
10
  && cd IndicTransToolkit \
11
  && pip install --editable . --use-pep517
12
-
13
  WORKDIR /code
14
 
 
 
 
 
15
  # Copy code
16
  COPY ./app /code/app
17
 
 
9
  RUN git clone https://github.com/VarunGumma/IndicTransToolkit \
10
  && cd IndicTransToolkit \
11
  && pip install --editable . --use-pep517
12
+
13
  WORKDIR /code
14
 
15
+ # Set HF cache to a writable directory
16
+ ENV HF_HOME=/code/app/hf_cache
17
+ ENV TRANSFORMERS_CACHE=/code/app/hf_cache
18
+
19
  # Copy code
20
  COPY ./app /code/app
21
 
app/main.py CHANGED
@@ -10,7 +10,6 @@ import os
10
  app = FastAPI()
11
 
12
  # app.mount("/static", StaticFiles(directory="app/static"), name="static")
13
- OPENL_API_KEY = os.getenv('OPENL_API_KEY')
14
 
15
  @app.get("/")
16
  async def root():
 
10
  app = FastAPI()
11
 
12
  # app.mount("/static", StaticFiles(directory="app/static"), name="static")
 
13
 
14
  @app.get("/")
15
  async def root():