Spaces:
Running
Running
Yash Chauhan
commited on
Commit
·
48e9676
1
Parent(s):
a06959c
[updated] docker image and req file
Browse files- Dockerfile +1 -1
- requirements.txt +2 -1
Dockerfile
CHANGED
@@ -1,4 +1,4 @@
|
|
1 |
-
FROM python:3.
|
2 |
|
3 |
# Set working directory
|
4 |
WORKDIR /app
|
|
|
1 |
+
FROM python:3.11-slim
|
2 |
|
3 |
# Set working directory
|
4 |
WORKDIR /app
|
requirements.txt
CHANGED
@@ -3,4 +3,5 @@ transformers
|
|
3 |
fastapi
|
4 |
uvicorn
|
5 |
pydantic
|
6 |
-
numpy
|
|
|
|
3 |
fastapi
|
4 |
uvicorn
|
5 |
pydantic
|
6 |
+
numpy
|
7 |
+
accelerate>=0.26.0
|