awacke1 commited on
Commit
3173aa5
1 Parent(s): cf02197

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +18 -1
app.py CHANGED
@@ -62,6 +62,22 @@ GEN_NAMES = [
62
  "huggingface/EleutherAI/gpt-j-6B",
63
  "huggingface/gpt2-large",
64
  ]
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
65
 
66
  #ASR
67
  def transcribe(audio):
@@ -119,7 +135,8 @@ def tts(text: str, model_name: str):
119
  return fp.name
120
 
121
  #Stories
122
- def Story(text: input):
 
123
  return
124
 
125
  #Blocks Rock It
 
62
  "huggingface/EleutherAI/gpt-j-6B",
63
  "huggingface/gpt2-large",
64
  ]
65
+ GENERATORS = {}
66
+ manager = ModelManager()
67
+ for MODEL_NAME in GENERATORS:
68
+ print(f"downloading {MODEL_NAME}")
69
+ model_path, config_path, model_item = manager.download_model(f"{MODEL_NAME}")
70
+ #vocoder_name: Optional[str] = model_item["default_vocoder"]
71
+ #vocoder_path = None
72
+ #vocoder_config_path = None
73
+ #if vocoder_name is not None:
74
+ # vocoder_path, vocoder_config_path, _ = manager.download_model(vocoder_name)
75
+
76
+ #synthesizer = Synthesizer(
77
+ # model_path, config_path, None, vocoder_path, vocoder_config_path,
78
+ #)
79
+ #MODELS[MODEL_NAME] = synthesizer
80
+
81
 
82
  #ASR
83
  def transcribe(audio):
 
135
  return fp.name
136
 
137
  #Stories
138
+ def Story(text: input, model_name: str):
139
+ generator =
140
  return
141
 
142
  #Blocks Rock It