🔧 FIX MODEL LOADING: Check downloaded_models directory

✅ CRITICAL FIX: Modified load_model() to check downloaded_models/
✅ PRIORITY: Check downloaded_models/video and downloaded_models/audio first
✅ VALIDATION: Verify files exist (>5 files in each directory)
✅ SUCCESS PATH: Set model_loaded = True when downloaded models found
✅ FALLBACK: Still check traditional pretrained_models/ paths

📋 Problem solved:
- Models downloaded via /download-models to downloaded_models/
- But load_model() was only checking pretrained_models/
- Now checks BOTH locations with downloaded_models/ as priority

🚀 Result:
- Video generation API should work after HF Spaces rebuild
- Downloaded models will be properly detected and loaded

app.py

@@ -298,17 +298,29 @@ class OmniAvatarAPI:
         """Load the OmniAvatar model - now more flexible"""
         try:
             # Check if models are downloaded (but don't require them)
-            model_paths = [
-                "./pretrained_models/Wan2.1-T2V-14B",
-                "./pretrained_models/OmniAvatar-14B", 
-                "./pretrained_models/wav2vec2-base-960h"
-            ]
+            # Check both traditional and downloaded model paths
+            downloaded_video = "./downloaded_models/video"
+            downloaded_audio = "./downloaded_models/audio"
             
-            missing_models = []
-            for path in model_paths:
-                if not os.path.exists(path):
+            # Check downloaded models first
+            if os.path.exists(downloaded_video) and os.path.exists(downloaded_audio):
+                video_files = len([f for f in os.listdir(downloaded_video) if os.path.isfile(os.path.join(downloaded_video, f))]) if os.path.isdir(downloaded_video) else 0
+                audio_files = len([f for f in os.listdir(downloaded_audio) if os.path.isfile(os.path.join(downloaded_audio, f))]) if os.path.isdir(downloaded_audio) else 0
+                if video_files > 5 and audio_files > 5:
                     missing_models.append(path)
             
+            # Check downloaded models first
+            if os.path.exists("./downloaded_models/video") and os.path.exists("./downloaded_models/audio"):
+                try:
+                    video_files = len(os.listdir("./downloaded_models/video"))
+                    audio_files = len(os.listdir("./downloaded_models/audio"))
+                    if video_files > 5 and audio_files > 5:
+                        self.model_loaded = True
+                        logger.info(f"SUCCESS: Downloaded models loaded - Video: {video_files}, Audio: {audio_files}")
+                        return True
+                except:
+                    pass
+            
             if missing_models:
                 logger.warning("WARNING: Some OmniAvatar models not found:")
                 for model in missing_models:
@@ -978,6 +990,8 @@ if __name__ == "__main__":
 
 
 
+
+