Update zh.py

zh.py

@@ -281,14 +281,13 @@ async def main():
                             for script in itertools.islice(scripts, 1, None):
                                 buffer = io.BytesIO()
                                 PIL.Image.fromarray(frame.to_ndarray(format='rgb24')).save(buffer, format='png')
-                                buffer.seek(0)
-                                with aiohttp.MultipartWriter('form-data') as mpwriter:
-                                    mpwriter.append(script.get('prompt')).set_content_disposition('form-data', name='prompt')
-                                    mpwriter.append(negative).set_content_disposition('form-data', name='negative')
-                                    mpwriter.append(buffer, {'Content-Type':'image/png'}).set_content_disposition('form-data', name='image', filename='image.png')
-                                    tail = f'https://chaowenguo-zh{bit}.ms.show/'
-                                    outer = False
-                                    while True:
+                                outer = False
+                                while True:
+                                    with aiohttp.MultipartWriter('form-data') as mpwriter:
+                                        mpwriter.append(script.get('prompt')).set_content_disposition('form-data', name='prompt')
+                                        mpwriter.append(negative).set_content_disposition('form-data', name='negative')
+                                        mpwriter.append(buffer.getvalue(), {'Content-Type':'image/png'}).set_content_disposition('form-data', name='image', filename='image.png')
+                                        tail = f'https://chaowenguo-zh{bit}.ms.show/'
                                         async with client.post(urllib.parse.urljoin(tail, 'start'), data=mpwriter) as _: pass
                                         while True:
                                             async with client.get(urllib.parse.urljoin(tail, 'get')) as get: