itod commited on
Commit
bc45070
1 Parent(s): d0ec710

run app directly

Browse files
Files changed (1) hide show
  1. Dockerfile +3 -3
Dockerfile CHANGED
@@ -7,7 +7,7 @@ ARG MODEL_NAME=WhereIsAI/UAE-Large-V1
7
  # Set the model name and path as environment variables
8
  ENV MODEL_NAME=$MODEL_NAME
9
  ENV MODEL_PATH=/models/${MODEL_NAME}
10
- ENV TRANSFORMERS_CACHE=/.cache
11
 
12
  # Create /.cache directory and make it writable
13
  RUN mkdir /.cache && chmod 777 /.cache
@@ -42,5 +42,5 @@ WORKDIR /app
42
  # Expose the FastAPI port
43
  EXPOSE 8080
44
 
45
- # Start the FastAPI server using the start script
46
- CMD ["/start.sh"]
 
7
  # Set the model name and path as environment variables
8
  ENV MODEL_NAME=$MODEL_NAME
9
  ENV MODEL_PATH=/models/${MODEL_NAME}
10
+ ENV HF_HOME=/.cache
11
 
12
  # Create /.cache directory and make it writable
13
  RUN mkdir /.cache && chmod 777 /.cache
 
42
  # Expose the FastAPI port
43
  EXPOSE 8080
44
 
45
+ # Instead of using start.sh, run the application directly
46
+ CMD ["python3", "/app/app.py"]