Commit
·
262ef16
1
Parent(s):
0b1744f
Update Dockerfile
Browse files- Dockerfile +6 -5
Dockerfile
CHANGED
@@ -10,17 +10,18 @@ COPY ./requirements.txt /code/requirements.txt
|
|
10 |
# Install requirements
|
11 |
RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
|
12 |
|
13 |
-
# Add your application code here (e.g., FastAPI code)
|
14 |
-
COPY main.py .
|
15 |
# Install transformers library
|
16 |
RUN pip install transformers
|
17 |
|
18 |
# Specify the model name from Hugging Face
|
19 |
-
ARG MODEL_NAME="gyesibiney/
|
20 |
|
21 |
# Download the model from Hugging Face (you can replace 'main' with a specific tag or version)
|
22 |
-
RUN transformers-cli login
|
23 |
-
|
|
|
|
|
|
|
24 |
|
25 |
# Add your application code here to use the downloaded model
|
26 |
|
|
|
10 |
# Install requirements
|
11 |
RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
|
12 |
|
|
|
|
|
13 |
# Install transformers library
|
14 |
RUN pip install transformers
|
15 |
|
16 |
# Specify the model name from Hugging Face
|
17 |
+
ARG MODEL_NAME="gyesibiney/Sentiment-review-analysis-roberta-3"
|
18 |
|
19 |
# Download the model from Hugging Face (you can replace 'main' with a specific tag or version)
|
20 |
+
RUN transformers-cli login
|
21 |
+
|
22 |
+
# Clone the model repository
|
23 |
+
RUN transformers-cli repo clone git lfs install
|
24 |
+
git clone https://huggingface.co/spaces/gyesibiney/sentiment-movie-review-FASTAPI-2 --path /code/model
|
25 |
|
26 |
# Add your application code here to use the downloaded model
|
27 |
|