gruhit-patel
commited on
Commit
•
a00bdea
1
Parent(s):
f33a809
Updated requirements
Browse files- backend.py +0 -1
- requirements.txt +3 -1
backend.py
CHANGED
@@ -1,4 +1,3 @@
|
|
1 |
-
import os
|
2 |
|
3 |
from model import get_model
|
4 |
|
|
|
|
|
1 |
|
2 |
from model import get_model
|
3 |
|
requirements.txt
CHANGED
@@ -1,6 +1,8 @@
|
|
1 |
streamlit
|
2 |
fastapi
|
3 |
uvicorn[standard]
|
4 |
-
|
|
|
|
|
5 |
pillow
|
6 |
numpy
|
|
|
1 |
streamlit
|
2 |
fastapi
|
3 |
uvicorn[standard]
|
4 |
+
torch==2.3.1
|
5 |
+
torchaudio==2.3.1
|
6 |
+
torchvision==0.18.1
|
7 |
pillow
|
8 |
numpy
|