Updated Gradio UI

app.py

@@ -52,23 +52,25 @@ def predict_accent(audio):
     label_map = {0: "Canadian English", 1: "England English"}
     return label_map[predicted_class_id]
 
-# Gradio Interface
-interface = gr.Interface(
-    fn=predict_accent,
-    inputs=gr.Audio(sources=["upload", "microphone"], type="filepath", label="🎤 Upload or Record Audio (WAV)"),
-    outputs=gr.Textbox(label="🗣️ Predicted Accent"),
-    title="🌍 Accent Classification App",
-    description="""
-This app uses a fine-tuned **Wav2Vec2** model to classify English accents as either **Canadian** or **England**.  
-It supports both audio upload and direct recording.
 
----
+# Custom UI with gr.Blocks
+with gr.Blocks() as demo:
+    gr.Markdown("## 🗣️ Accent Classification App")
+    gr.Markdown("This app classifies English accents as either **Canadian** or **England** using a fine-tuned Wav2Vec2 model.")
+    
+    with gr.Row():
+        with gr.Column():
+            audio_input = gr.Audio(sources=["upload", "microphone"], type="filepath", label="Upload or Record Audio (WAV)")
+            predict_button = gr.Button("Predict")
+        with gr.Column():
+            result_output = gr.Textbox(label="Predicted Accent")
 
-👨‍💻 **Author**: [Anand Purushottam](https://www.linkedin.com/in/creativepurus/)  
-🔗 [GitHub: creativepurus](https://github.com/creativepurus)
+    predict_button.click(fn=predict_accent, inputs=audio_input, outputs=result_output)
 
-""",
-    allow_flagging="never"
-)
+    gr.Markdown("---")
+    gr.Markdown("""
+    👨‍💻 Created by **Anand Purushottam**  
+    🔗 [GitHub](https://github.com/creativepurus) | [LinkedIn](https://linkedin.com/in/creativepurus)
+    """)
 
-interface.launch()
+demo.launch()