Spaces:
Sleeping
Sleeping
simplify dockerfile
Browse files- Dockerfile +10 -9
- app.py +0 -4
- requirements.txt +1 -1
Dockerfile
CHANGED
@@ -1,13 +1,14 @@
|
|
1 |
FROM python:3.9
|
2 |
|
3 |
-
WORKDIR /
|
4 |
-
COPY ./requirements.txt /
|
5 |
-
RUN pip install --no-cache-dir
|
6 |
-
RUN useradd -m -u 1000 user
|
7 |
-
USER user
|
8 |
-
ENV HOME=/home/user \
|
9 |
-
PATH=/home/user/.local/bin:$PATH
|
10 |
-
WORKDIR $HOME/app
|
11 |
-
COPY --chown=user . $HOME/app
|
|
|
12 |
|
13 |
CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]
|
|
|
1 |
FROM python:3.9
|
2 |
|
3 |
+
WORKDIR /src
|
4 |
+
COPY ./requirements.txt /src/requirements.txt
|
5 |
+
RUN pip install --no-cache-dir -r /code/requirements.txt
|
6 |
+
#RUN useradd -m -u 1000 user
|
7 |
+
#USER user
|
8 |
+
#ENV HOME=/home/user \
|
9 |
+
# PATH=/home/user/.local/bin:$PATH
|
10 |
+
#WORKDIR $HOME/app
|
11 |
+
#COPY --chown=user . $HOME/app
|
12 |
+
EXPOSE 7860
|
13 |
|
14 |
CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]
|
app.py
CHANGED
@@ -9,7 +9,3 @@ pipe = pipeline("text2text-generation", model="google/flan-t5-small")
|
|
9 |
def generate(text: str):
|
10 |
output = pipe(text)
|
11 |
return {"output": output[0]["generated_text"]}
|
12 |
-
|
13 |
-
# serve the app
|
14 |
-
# if __name__ == "__main__":
|
15 |
-
# uvicorn.run(app, host="0.0.0.0", port=7860)
|
|
|
9 |
def generate(text: str):
|
10 |
output = pipe(text)
|
11 |
return {"output": output[0]["generated_text"]}
|
|
|
|
|
|
|
|
requirements.txt
CHANGED
@@ -2,5 +2,5 @@ fastapi==0.74.*
|
|
2 |
requests==2.27.*
|
3 |
uvicorn[standard]==0.17.*
|
4 |
sentencepiece==0.1.*
|
5 |
-
torch==1.11.*
|
6 |
transformers==4.*
|
|
|
2 |
requests==2.27.*
|
3 |
uvicorn[standard]==0.17.*
|
4 |
sentencepiece==0.1.*
|
5 |
+
#torch==1.11.*
|
6 |
transformers==4.*
|