Spaces:
Sleeping
Sleeping
Commit ·
5cce2f0
1
Parent(s): 4de5872
Fix audio model loading with detailed logging
Browse files- src/main.py +20 -5
src/main.py
CHANGED
|
@@ -139,14 +139,29 @@ async def load_models():
|
|
| 139 |
|
| 140 |
# --- Load Audio Model (HuggingFace Transformers) ---
|
| 141 |
try:
|
| 142 |
-
from transformers import pipeline
|
| 143 |
print("Loading Audio Model (Deepfake-audio-detection)...")
|
| 144 |
-
|
| 145 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 146 |
except Exception as e:
|
| 147 |
-
print(f"
|
|
|
|
|
|
|
| 148 |
|
| 149 |
-
|
|
|
|
|
|
|
| 150 |
|
| 151 |
# --- 3. Define API Endpoints ---
|
| 152 |
|
|
|
|
| 139 |
|
| 140 |
# --- Load Audio Model (HuggingFace Transformers) ---
|
| 141 |
try:
|
|
|
|
| 142 |
print("Loading Audio Model (Deepfake-audio-detection)...")
|
| 143 |
+
print("Importing transformers pipeline...")
|
| 144 |
+
from transformers import pipeline
|
| 145 |
+
print("Pipeline imported successfully.")
|
| 146 |
+
|
| 147 |
+
print("Downloading and loading audio classification model...")
|
| 148 |
+
models["audio_pipeline"] = pipeline(
|
| 149 |
+
"audio-classification",
|
| 150 |
+
model="motheecreator/Deepfake-audio-detection",
|
| 151 |
+
device="cpu"
|
| 152 |
+
)
|
| 153 |
+
print("Audio Model loaded successfully!")
|
| 154 |
+
except ImportError as ie:
|
| 155 |
+
print(f"CRITICAL: transformers import failed: {ie}")
|
| 156 |
+
print("Please install: pip install transformers")
|
| 157 |
except Exception as e:
|
| 158 |
+
print(f"CRITICAL: Failed to load Audio Model: {type(e).__name__}: {e}")
|
| 159 |
+
import traceback
|
| 160 |
+
traceback.print_exc()
|
| 161 |
|
| 162 |
+
# Print final status
|
| 163 |
+
print("--- Model loading complete ---")
|
| 164 |
+
print(f"Loaded models: {list(models.keys())}")
|
| 165 |
|
| 166 |
# --- 3. Define API Endpoints ---
|
| 167 |
|