.

src/classifier.py

@@ -12,14 +12,14 @@ class MusicGenreClassifier:
             self.text_classifier = pipeline(
                 "zero-shot-classification",
                 model="facebook/bart-large-mnli",
-                device_map="auto"
+                device="cpu"
             )
             
             # For audio classification, we'll use MIT's music classification model
             self.audio_classifier = pipeline(
                 "audio-classification",
                 model="mit/ast-finetuned-audioset-10-10-0.4593",
-                device_map="auto"
+                device="cpu"
             )
         except Exception as e:
             print(f"Warning: GPU initialization failed, falling back to CPU. Error: {str(e)}")