Update app.py

app.py

@@ -3,12 +3,12 @@ import torch
 import torchaudio
 import os
 import tempfile
-from huggingface_hub import hf_hub_download # 1. Import the downloader
+from huggingface_hub import hf_hub_download
 from sam_audio import SAMAudio, SAMAudioProcessor
 
 # --- Initialization ---
 device = "cuda" if torch.cuda.is_available() else "cpu"
-REPO_ID = "your-username/your-model-name" 
+REPO_ID = "ray-006/model-audio" 
 checkpoint_path = hf_hub_download(repo_id=REPO_ID, filename="checkpoint.pt")
 
 # 4. Load the model and apply your weights