Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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
|