()
| 177 | |
| 178 | |
| 179 | def main(): |
| 180 | args = parse_arguments() |
| 181 | |
| 182 | # Main infer params |
| 183 | body = { |
| 184 | "text": args.text, |
| 185 | "stream": args.stream, |
| 186 | "lang": args.lang, |
| 187 | "filename": args.filename, |
| 188 | "skip_refine_text": args.skip_refine_text, |
| 189 | "refine_text_only": args.refine_text_only, |
| 190 | "use_decoder": args.use_decoder, |
| 191 | "audio_seed": args.audio_seed, |
| 192 | "text_seed": args.text_seed, |
| 193 | "do_text_normalization": args.do_text_normalization, |
| 194 | "do_homophone_replacement": args.do_homophone_replacement, |
| 195 | } |
| 196 | # Refinement text parameters |
| 197 | params_refine_text = { |
| 198 | "prompt": args.refine_prompt, |
| 199 | "top_P": args.refine_top_P, |
| 200 | "top_K": args.refine_top_K, |
| 201 | "temperature": args.refine_temperature, |
| 202 | "repetition_penalty": args.refine_repetition_penalty, |
| 203 | "max_new_token": args.refine_max_new_token, |
| 204 | "min_new_token": args.refine_min_new_token, |
| 205 | "show_tqdm": args.refine_show_tqdm, |
| 206 | "ensure_non_empty": args.refine_ensure_non_empty, |
| 207 | "stream_batch": args.refine_stream_batch, |
| 208 | } |
| 209 | body["params_refine_text"] = params_refine_text |
| 210 | |
| 211 | # Infer code parameters |
| 212 | params_infer_code = { |
| 213 | "prompt": args.infer_prompt, |
| 214 | "top_P": args.infer_top_P, |
| 215 | "top_K": args.infer_top_K, |
| 216 | "temperature": args.infer_temperature, |
| 217 | "repetition_penalty": args.infer_repetition_penalty, |
| 218 | "max_new_token": args.infer_max_new_token, |
| 219 | "min_new_token": args.infer_min_new_token, |
| 220 | "show_tqdm": args.infer_show_tqdm, |
| 221 | "ensure_non_empty": args.infer_ensure_non_empty, |
| 222 | "stream_batch": args.infer_stream_batch, |
| 223 | "spk_emb": args.infer_spk_emb, |
| 224 | } |
| 225 | body["params_infer_code"] = params_infer_code |
| 226 | |
| 227 | try: |
| 228 | response = requests.post(CHATTTS_URL, json=body) |
| 229 | response.raise_for_status() |
| 230 | with zipfile.ZipFile(BytesIO(response.content), "r") as zip_ref: |
| 231 | tgt = args.tgt |
| 232 | # filename=args.filename |
| 233 | os.makedirs(tgt, exist_ok=True) |
| 234 | zip_ref.extractall(tgt) |
| 235 | print(f"Extracted files:{tgt}/{filename}") |
| 236 | # print(tgt) |
no test coverage detected