ray-006 commited on
Commit
d489a04
·
verified ·
1 Parent(s): 19e6abf

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -2
app.py CHANGED
@@ -7,8 +7,10 @@ from huggingface_hub import hf_hub_download
7
  from sam_audio import SAMAudio, SAMAudioProcessor
8
 
9
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
10
- model = SAMAudio.from_pretrained("facebook/sam-audio-large").to(device).eval()
11
- processor = SAMAudioProcessor.from_pretrained("facebook/sam-audio-large")
 
 
12
 
13
  def separate_audio(audio_path, description, reranking_candidates):
14
  if audio_path is None or not description:
 
7
  from sam_audio import SAMAudio, SAMAudioProcessor
8
 
9
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
10
+ model = SAMAudio.from_pretrained("facebook/sam-audio-large",
11
+ token=os.environ.get("HF_TOKEN")).to(device).eval()
12
+ processor = SAMAudioProcessor.from_pretrained("facebook/sam-audio-large",
13
+ token=os.environ.get("HF_TOKEN"))
14
 
15
  def separate_audio(audio_path, description, reranking_candidates):
16
  if audio_path is None or not description: