youngshen commited on
Commit
a287ea4
·
verified ·
1 Parent(s): 5f884be

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -32,7 +32,7 @@ def predict(text, speaker):
32
  if Interest == "":
33
  inputs = processor(text=getNews(Interest),
34
  return_tensors="pt")
35
- else
36
  inputs = processor(text=getNews(text),
37
  return_tensors="pt")
38
 
@@ -41,7 +41,6 @@ def predict(text, speaker):
41
  input_ids = input_ids[..., :model.config.max_text_positions]
42
 
43
  speaker_embedding = np.load("spkemb/cmu_us_bdl_arctic-wav-arctic_a0009.npy")
44
-
45
  speaker_embedding = torch.tensor(speaker_embedding).unsqueeze(0)
46
 
47
  speech = model.generate_speech(input_ids, speaker_embedding, vocoder=vocoder)
 
32
  if Interest == "":
33
  inputs = processor(text=getNews(Interest),
34
  return_tensors="pt")
35
+ else:
36
  inputs = processor(text=getNews(text),
37
  return_tensors="pt")
38
 
 
41
  input_ids = input_ids[..., :model.config.max_text_positions]
42
 
43
  speaker_embedding = np.load("spkemb/cmu_us_bdl_arctic-wav-arctic_a0009.npy")
 
44
  speaker_embedding = torch.tensor(speaker_embedding).unsqueeze(0)
45
 
46
  speech = model.generate_speech(input_ids, speaker_embedding, vocoder=vocoder)