Diamanta commited on
Commit
7357a31
·
verified ·
1 Parent(s): cc97c64
Files changed (2) hide show
  1. Dockerfile +3 -7
  2. start.sh +6 -0
Dockerfile CHANGED
@@ -24,10 +24,6 @@ COPY . /app
24
  ENV MODEL_REPO=TheBloke/phi-2-GGUF
25
  ENV MODEL_FILE=phi-2.Q4_K_M.gguf
26
 
27
- # Create model loader script
28
- RUN echo '#!/bin/bash\n'\
29
- 'python download_model.py\n'\
30
- 'uvicorn app:app --host 0.0.0.0 --port 7860' > entrypoint.sh && \
31
- chmod +x entrypoint.sh
32
-
33
- CMD ["./entrypoint.sh"]
 
24
  ENV MODEL_REPO=TheBloke/phi-2-GGUF
25
  ENV MODEL_FILE=phi-2.Q4_K_M.gguf
26
 
27
+ COPY start.sh .
28
+ RUN chmod +x start.sh
29
+ CMD ["./start.sh"]
 
 
 
 
start.sh ADDED
@@ -0,0 +1,6 @@
 
 
 
 
 
 
 
1
+ #!/bin/bash
2
+ echo "Downloading model..."
3
+ python download_model.py
4
+
5
+ echo "Starting API server..."
6
+ uvicorn app:app --host 0.0.0.0 --port 7860