fix AttributeError with .shape

app.py

@@ -15,13 +15,20 @@ dialect_mapping = {
     "Maghrebi": "Maghrebi Arabic"
 }
 
-def predict_dialect(audio, sr):
+def predict_dialect(audio):
+    # The audio input from Gradio is a tuple of (sample_rate, audio_array)
+    if isinstance(audio, tuple) and len(audio) == 2:
+        sr, audio_array = audio
+    else:
+        # Handle error case
+        return {"Error": 1.0}
+    
     # Process the audio input
-    if len(audio.shape) > 1:
-        audio = audio.mean(axis=1)  # Convert stereo to mono
+    if len(audio_array.shape) > 1:
+        audio_array = audio_array.mean(axis=1)  # Convert stereo to mono
     
     # Classify the dialect
-    predictions = classifier({"sampling_rate": sr, "raw": audio})
+    predictions = classifier({"sampling_rate": sr, "raw": audio_array})
     
     # Format results for display
     results = {}
@@ -34,7 +41,7 @@ def predict_dialect(audio, sr):
 # Create the Gradio interface
 demo = gr.Interface(
     fn=predict_dialect,
-    inputs=gr.Audio(type="numpy", label="Upload or Record Audio"),
+    inputs=gr.Audio(),  # Simplified audio input
     outputs=gr.Label(num_top_classes=5, label="Predicted Dialect"),
     title="Arabic Dialect Identifier",
     description="""This demo identifies Arabic dialects from speech audio.
@@ -49,4 +56,4 @@ demo = gr.Interface(
 )
 
 # Launch the app
-demo.launch()
+demo.launch()