Commit ·
813d818
1
Parent(s): d1b31ce
Summary
Browse files- app/model.py +0 -2
- requirements.txt +2 -1
app/model.py
CHANGED
|
@@ -17,7 +17,6 @@ from app.config import config_data
|
|
| 17 |
|
| 18 |
def load_model(model_url, model_path):
|
| 19 |
try:
|
| 20 |
-
# Загрузка модели
|
| 21 |
with requests.get(model_url, stream=True) as response:
|
| 22 |
with open(model_path, "wb") as file:
|
| 23 |
for chunk in response.iter_content(chunk_size=8192):
|
|
@@ -28,7 +27,6 @@ def load_model(model_url, model_path):
|
|
| 28 |
return None
|
| 29 |
|
| 30 |
|
| 31 |
-
# Загрузите модель
|
| 32 |
pth_model = load_model(config_data.model_url, config_data.model_path)
|
| 33 |
|
| 34 |
|
|
|
|
| 17 |
|
| 18 |
def load_model(model_url, model_path):
|
| 19 |
try:
|
|
|
|
| 20 |
with requests.get(model_url, stream=True) as response:
|
| 21 |
with open(model_path, "wb") as file:
|
| 22 |
for chunk in response.iter_content(chunk_size=8192):
|
|
|
|
| 27 |
return None
|
| 28 |
|
| 29 |
|
|
|
|
| 30 |
pth_model = load_model(config_data.model_url, config_data.model_path)
|
| 31 |
|
| 32 |
|
requirements.txt
CHANGED
|
@@ -4,4 +4,5 @@ torch==2.1.2
|
|
| 4 |
torchaudio==2.1.2
|
| 5 |
torchvision==0.16.2
|
| 6 |
mediapipe==0.10.9
|
| 7 |
-
pillow==10.2.0
|
|
|
|
|
|
| 4 |
torchaudio==2.1.2
|
| 5 |
torchvision==0.16.2
|
| 6 |
mediapipe==0.10.9
|
| 7 |
+
pillow==10.2.0
|
| 8 |
+
toml==0.10.2
|