Commit Β·
d4e4e83
1
Parent(s): 9b5e433
Add packages.txt and auto-download model for cloud deployment
Browse files- app.py +10 -5
- packages.txt +2 -0
app.py
CHANGED
|
@@ -172,13 +172,18 @@ _init_state()
|
|
| 172 |
# Helpers
|
| 173 |
# βββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββ
|
| 174 |
def check_model_available():
|
| 175 |
-
"""Return (available: bool, message: str)."""
|
| 176 |
model_path = os.path.join(BASE_DIR, "models", "arcface.onnx")
|
| 177 |
if not os.path.exists(model_path):
|
| 178 |
-
|
| 179 |
-
|
| 180 |
-
"
|
| 181 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 182 |
return True, ""
|
| 183 |
|
| 184 |
|
|
|
|
| 172 |
# Helpers
|
| 173 |
# βββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββ
|
| 174 |
def check_model_available():
|
| 175 |
+
"""Return (available: bool, message: str). Auto-downloads if missing."""
|
| 176 |
model_path = os.path.join(BASE_DIR, "models", "arcface.onnx")
|
| 177 |
if not os.path.exists(model_path):
|
| 178 |
+
try:
|
| 179 |
+
import download_model
|
| 180 |
+
with st.spinner("Downloading ArcFace ONNX model (~166MB). Please wait..."):
|
| 181 |
+
try:
|
| 182 |
+
download_model.download_direct_onnx()
|
| 183 |
+
except Exception:
|
| 184 |
+
download_model.download_zip_and_extract()
|
| 185 |
+
except Exception as e:
|
| 186 |
+
return False, f"β **ArcFace model download failed.** \nError: {e}"
|
| 187 |
return True, ""
|
| 188 |
|
| 189 |
|
packages.txt
ADDED
|
@@ -0,0 +1,2 @@
|
|
|
|
|
|
|
|
|
|
| 1 |
+
libgl1-mesa-glx
|
| 2 |
+
libglib2.0-0
|