piyushnaula commited on
Commit
5cce2f0
·
1 Parent(s): 4de5872

Fix audio model loading with detailed logging

Browse files
Files changed (1) hide show
  1. 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
- models["audio_pipeline"] = pipeline("audio-classification", model="motheecreator/Deepfake-audio-detection")
145
- print("Audio Model loaded successfully.")
 
 
 
 
 
 
 
 
 
 
 
 
146
  except Exception as e:
147
- print(f"WARNING: Failed to load Audio Model: {e}")
 
 
148
 
149
- print("--- All models loaded. API is ready. ---")
 
 
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