Norphel commited on
Commit
172e398
·
verified ·
1 Parent(s): 13765c8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -7,7 +7,7 @@ import re
7
 
8
  @st.cache_resource
9
  def load_model(gender):
10
- model_path = './tts/male' if gender == 'Male' else './tts/female'
11
  processor = VitsTokenizer.from_pretrained(model_path)
12
  model = VitsModel.from_pretrained(model_path)
13
  return model, processor
 
7
 
8
  @st.cache_resource
9
  def load_model(gender):
10
+ model_path = 'ingye/dzo_tts_v2' if gender == 'Male' else 'ingye/dzo_tts_v2'
11
  processor = VitsTokenizer.from_pretrained(model_path)
12
  model = VitsModel.from_pretrained(model_path)
13
  return model, processor