Add delay and verification after model download to handle timing issues

app.py

@@ -46,6 +46,15 @@ def ensure_weights():
         allow_patterns=["**/*"],
         token=os.getenv("HF_TOKEN") or os.getenv("HUGGINGFACE_TOKEN"),
     )
+    
+    # Add a small delay to ensure files are fully written
+    import time
+    time.sleep(2)
+    
+    # Verify critical files exist
+    for file_path in need:
+        if not os.path.exists(file_path):
+            logger.warning("File %s still not found after download", file_path)
 
 
 def get_runner() -> SimpleRunner: