Update app.py

app.py

@@ -9,6 +9,14 @@ print("Loading model...")
 model_id = "badrex/mms-300m-arabic-dialect-identifier"
 classifier = pipeline("audio-classification", model=model_id)
 print("Model loaded successfully")
+classifier.to('cuda')
+print("Model moved to GPU successfully")
+
+
+@spaces.GPU
+def generate(audio_segment, sr=16000):
+    return classifier({"sampling_rate": sr, "raw": audio_segment})
+
 
 # Define dialect mapping
 dialect_mapping = {