Upload folder using huggingface_hub

app.py

@@ -10,6 +10,8 @@ matplotlib.use("Agg")
 import matplotlib.pyplot as plt
 import numpy as np
 
+DATASET_URL = "https://huggingface.co/datasets/mandipgoswami/AnomalyMachine-50K"
+
 DATASET_INFO = {
     "total_clips": 50000,
     "machines": ["fan", "pump", "compressor", "conveyor_belt", "electric_motor", "valve"],
@@ -57,6 +59,12 @@ def predict_anomaly(audio_file, machine_type):
     if audio_file is None:
         return "Please upload an audio file.", None, None
     
+    # Gradio Audio can return dict with "path" key in some versions
+    if isinstance(audio_file, dict):
+        audio_file = audio_file.get("path") or audio_file.get("name")
+    if not audio_file:
+        return "Please upload an audio file.", None, None
+    
     import random
     random.seed(hash(str(audio_file)) % 1000)
     is_anomaly = random.random() > 0.5
@@ -91,7 +99,7 @@ with gr.Blocks(theme=theme, title="AnomalyMachine-50K Demo") as app:
     <div style="text-align: center; padding: 20px;">
         <h1>🏭 AnomalyMachine-50K</h1>
         <p>Synthetic Industrial Machine Sound Anomaly Detection Dataset</p>
-        <p><a href="https://huggingface.co/datasets/AnomalyMachine-50K" target="_blank">View Dataset →</a></p>
+        <p><a href="https://huggingface.co/datasets/mandipgoswami/AnomalyMachine-50K" target="_blank">View Dataset →</a></p>
     </div>
     """)
     
@@ -113,7 +121,8 @@ with gr.Blocks(theme=theme, title="AnomalyMachine-50K Demo") as app:
             predict_btn.click(
                 fn=predict_anomaly,
                 inputs=[audio_input, machine_dropdown],
-                outputs=[result_html, input_spec, ref_spec]
+                outputs=[result_html, input_spec, ref_spec],
+                api_name="predict",
             )
         
         with gr.Tab("📊 Explore Dataset"):
@@ -127,7 +136,7 @@ with gr.Blocks(theme=theme, title="AnomalyMachine-50K Demo") as app:
                     <li>Normal: {DATASET_INFO['normal_ratio']:.0%} | Anomalous: {DATASET_INFO['anomalous_ratio']:.0%}</li>
                 </ul>
                 <p style="text-align: center;">
-                    <a href="https://huggingface.co/datasets/AnomalyMachine-50K" target="_blank">
+                    <a href="https://huggingface.co/datasets/mandipgoswami/AnomalyMachine-50K" target="_blank">
                         <button style="padding: 10px 20px; font-size: 16px;">📥 Download Dataset</button>
                     </a>
                 </p>
@@ -138,7 +147,7 @@ with gr.Blocks(theme=theme, title="AnomalyMachine-50K Demo") as app:
     <div style="text-align: center; padding: 20px; border-top: 1px solid #333; margin-top: 40px;">
         <p style="color: #888; font-size: 14px;">
             License: CC-BY 4.0 | 
-            <a href="https://huggingface.co/datasets/AnomalyMachine-50K" target="_blank">Dataset</a> | 
+            <a href="https://huggingface.co/datasets/mandipgoswami/AnomalyMachine-50K" target="_blank">Dataset</a> | 
             <a href="https://github.com/mandip42/anomaly-machine-50k" target="_blank">GitHub</a>
         </p>
     </div>