ray-006 commited on
Commit
6bdfb90
·
verified ·
1 Parent(s): a8bada1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -9,8 +9,7 @@ from sam_audio import SAMAudio, SAMAudioProcessor
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:
 
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
 
14
  def separate_audio(audio_path, description, reranking_candidates):
15
  if audio_path is None or not description: