Update zh.py

zh.py

@@ -93,7 +93,7 @@ async def main():
                             video.width = reader.streams.video[0].width
                             video.height = reader.streams.video[0].height
                             video.pix_fmt = 'yuv420p'
-                            with av.open(io.BytesIO(await sys.modules['__main__'].tts(scripts[0]))) as container:
+                            with av.open(io.BytesIO(await sys.modules['__main__'].tts(scripts[0])), format='mp3') as container:
                                 audio = writer.add_stream('aac', rate=container.streams.audio[0].rate)
                                 speed = builtins.float(container.streams.audio[0].duration * container.streams.audio[0].time_base) / 10
                                 print(speed)
@@ -147,7 +147,7 @@ async def main():
                                                         case 200:
                                                             with zipfile.ZipFile(io.BytesIO(await get.content.read())) as archive: 
                                                                 with av.open(io.BytesIO(archive.read('output.mp4'))) as reader:
-                                                                    with av.open(io.BytesIO(await sys.modules['__main__'].tts(script))) as container:
+                                                                    with av.open(io.BytesIO(await sys.modules['__main__'].tts(script)), format='mp3') as container:
                                                                         speed = builtins.float(container.streams.audio[0].duration * container.streams.audio[0].time_base) / 10
                                                                         print(speed)
                                                                         graph = av.filter.Graph()
@@ -184,6 +184,7 @@ async def main():
                             writer.mux(video.encode())
                             writer.mux(audio.encode())
                     break
+    huggingface_hub.upload_file(path_or_fileobj='output.mp4', path_in_repo='zh.mp4', repo_id='chaowenguo/wan', repo_type='dataset', token=os.getenv('huggingface'))
     await asyncio.sleep(math.inf)
 
 uvloop.run(main())