Hadeel11 commited on
Commit
81ef716
1 Parent(s): d7c7a6c

Fix Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +2 -2
Dockerfile CHANGED
@@ -9,8 +9,8 @@ RUN pip install transformers
9
  ENV MODEL_NAME="Hadeel11/your-finetuned-model"
10
  ENV USE_FP16=True
11
 
12
- Expose the port for the inference API
13
  EXPOSE 8080
14
 
15
- Run the TGI server
16
  CMD ["python3", "-m", "transformers_serve", "--model-id", "${MODEL_NAME}"]
 
9
  ENV MODEL_NAME="Hadeel11/your-finetuned-model"
10
  ENV USE_FP16=True
11
 
12
+
13
  EXPOSE 8080
14
 
15
+
16
  CMD ["python3", "-m", "transformers_serve", "--model-id", "${MODEL_NAME}"]