Copy PANNs to multiple locations

api.py

@@ -8,11 +8,8 @@ import gradio as gr
 
 sys.path.insert(0, os.path.dirname(os.path.abspath(__file__)))
 
-# Set PANNs path BEFORE importing anything else
-os.environ["PANNS_MODELS_PATH"] = "/root/panns_data"
-PANNS_CHECKPOINT = "/root/panns_data/Cnn14_mAP=0.431.pth"
-
-os.makedirs("/root/panns_data", exist_ok=True)
+# Download PANNs checkpoint to current directory
+PANNS_CHECKPOINT = "Cnn14_mAP=0.431.pth"
 
 if not os.path.exists(PANNS_CHECKPOINT):
     print("📥 Downloading PANNs checkpoint (~320MB)...")
@@ -21,10 +18,17 @@ if not os.path.exists(PANNS_CHECKPOINT):
         "https://zenodo.org/record/3987831/files/Cnn14_mAP%3D0.431.pth?download=1",
         PANNS_CHECKPOINT
     )
-    print("✅ PANNs checkpoint ready!")
+    print(f"✅ PANNs checkpoint ready! Size: {os.path.getsize(PANNS_CHECKPOINT)}")
 else:
     print(f"✅ PANNs checkpoint exists: {os.path.getsize(PANNS_CHECKPOINT)} bytes")
 
+# Also copy to where panns_inference might look
+os.makedirs("/root/panns_data", exist_ok=True)
+if not os.path.exists("/root/panns_data/Cnn14_mAP=0.431.pth"):
+    import shutil
+    shutil.copy(PANNS_CHECKPOINT, "/root/panns_data/Cnn14_mAP=0.431.pth")
+    print("✅ Copied to /root/panns_data/")
+
 DB_PATH = "library/snare_scout.sqlite"
 if not os.path.exists(DB_PATH):
     import requests