trretretret commited on
Commit
d4e4e83
Β·
1 Parent(s): 9b5e433

Add packages.txt and auto-download model for cloud deployment

Browse files
Files changed (2) hide show
  1. app.py +10 -5
  2. 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
- return False, (
179
- "❌ **ArcFace model not found.** \n"
180
- "Run `python download_model.py` in the project directory."
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