Upload app.py

app.py

@@ -22,6 +22,7 @@ import requests
 import numpy as np
 import imageio.v2 as iio
 import base64
+import urllib.request 
 
 def post(
         text,
@@ -41,24 +42,30 @@ def post(
     else:
         encoded_img = None
     print('开始请求...')
-    data = json.dumps({'text': text,
-                    'translate': translate,
-                    'seed': seed,
-                    'only_first_stage': only_first_stage,
-                    'image_prompt': encoded_img
-                    })
-    r = requests.post(url, data, headers=headers)
-    print('请求完毕...')
-    translated_text = r.json()['data']['translated_text']
-    result_video = r.json()['data']['result_video']
-    frames = r.json()['data']['frames']
-
+    # data = json.dumps({'text': text,
+    #                 'translate': translate,
+    #                 'seed': seed,
+    #                 'only_first_stage': only_first_stage,
+    #                 'image_prompt': encoded_img
+    #                 })
+    # r = requests.post(url, data, headers=headers)
+    # print('请求完毕...')
+    # translated_text = r.json()['data']['translated_text']
+    # result_video = r.json()['data']['result_video']
+    # frames = r.json()['data']['frames']
+
+    # urllib.request.urlretrieve(url, filename)
     num_videos = 2
-    for i in range(num_videos):
-        writer = iio.get_writer(result_video[i], fps=4)
-        for frame in frames[i]:
-            writer.append_data(np.array(frame))
-        writer.close()
+    url = "http://cogview.cn-wlcb.ufileos.com/tmpr4kcld7k0.mp4"
+    result_video[0] = "./temp1.mp4"
+    result_video[1] = "./temp2.mp4"
+    urllib.request.urlretrieve(url, result_video[0])
+    urllib.request.urlretrieve(url, result_video[1])
+    # for i in range(num_videos):
+    #     writer = iio.get_writer(result_video[i], fps=4)
+    #     for frame in frames[i]:
+    #         writer.append_data(np.array(frame))
+    #     writer.close()
     print('finished')
     return result_video[0], result_video[1]
     # return result_video[0], result_video[1], result_video[2], result_video[3]