Spaces:
Running
Running
Andrey
commited on
Commit
•
7c4b4ea
1
Parent(s):
d0a6fe9
Update requirements.txt (#15)
Browse files* Update requirements.txt
* Update formatting.
* Update formatting.
* Update formatting.
- .pre-commit-config.yaml +1 -1
- requirements.txt +2 -1
- src/ml_utils.py +0 -1
.pre-commit-config.yaml
CHANGED
|
@@ -6,7 +6,7 @@ repos:
|
|
| 6 |
- id: end-of-file-fixer
|
| 7 |
- id: trailing-whitespace
|
| 8 |
- repo: https://github.com/psf/black
|
| 9 |
-
rev: '
|
| 10 |
hooks:
|
| 11 |
- id: black
|
| 12 |
args: [--config=pyproject.toml]
|
|
|
|
| 6 |
- id: end-of-file-fixer
|
| 7 |
- id: trailing-whitespace
|
| 8 |
- repo: https://github.com/psf/black
|
| 9 |
+
rev: '23.7.0'
|
| 10 |
hooks:
|
| 11 |
- id: black
|
| 12 |
args: [--config=pyproject.toml]
|
requirements.txt
CHANGED
|
@@ -7,8 +7,9 @@ opencv-python-headless>=4.5.3.56
|
|
| 7 |
pandas==1.5.1
|
| 8 |
Pillow==9.2.0
|
| 9 |
rich==12.6.0
|
| 10 |
-
streamlit==1.
|
| 11 |
streamlit_drawable_canvas==0.9.2
|
| 12 |
tomli==2.0.1
|
| 13 |
torchvision
|
|
|
|
| 14 |
https://download.pytorch.org/whl/cpu/torch-1.13.1%2Bcpu-cp310-cp310-linux_x86_64.whl
|
|
|
|
| 7 |
pandas==1.5.1
|
| 8 |
Pillow==9.2.0
|
| 9 |
rich==12.6.0
|
| 10 |
+
streamlit==1.26.0
|
| 11 |
streamlit_drawable_canvas==0.9.2
|
| 12 |
tomli==2.0.1
|
| 13 |
torchvision
|
| 14 |
+
altair==5.0.1
|
| 15 |
https://download.pytorch.org/whl/cpu/torch-1.13.1%2Bcpu-cp310-cp310-linux_x86_64.whl
|
src/ml_utils.py
CHANGED
|
@@ -198,7 +198,6 @@ def predict(
|
|
| 198 |
|
| 199 |
@st.cache
|
| 200 |
def get_model():
|
| 201 |
-
|
| 202 |
model_name = 'model_files/best_model.pth'
|
| 203 |
|
| 204 |
model = Net()
|
|
|
|
| 198 |
|
| 199 |
@st.cache
|
| 200 |
def get_model():
|
|
|
|
| 201 |
model_name = 'model_files/best_model.pth'
|
| 202 |
|
| 203 |
model = Net()
|