yufengzhu commited on
Commit
b85d618
·
1 Parent(s): ba2e4dd

Upload app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -2
app.py CHANGED
@@ -41,14 +41,16 @@ def post(
41
  'translate': translate,
42
  'seed': seed,
43
  'only_first_stage': only_first_stage,
44
- 'image_prompt': encoded_img
45
  })
46
  r = requests.post(url, data, headers=headers)
47
 
48
  translated_text = r.json()['data']['translated_text']
49
  result_video = r.json()['data']['result_video']
50
  frames = r.json()['data']['frames']
51
- for i in range(2):
 
 
52
  writer = iio.get_writer(result_video[i], fps=4)
53
  for frame in frames[i]:
54
  writer.append_data(np.array(frame))
 
41
  'translate': translate,
42
  'seed': seed,
43
  'only_first_stage': only_first_stage,
44
+ 'image_prompt': str(encoded_img, encoding='utf-8')
45
  })
46
  r = requests.post(url, data, headers=headers)
47
 
48
  translated_text = r.json()['data']['translated_text']
49
  result_video = r.json()['data']['result_video']
50
  frames = r.json()['data']['frames']
51
+
52
+ num_videos = 2
53
+ for i in range(num_videos):
54
  writer = iio.get_writer(result_video[i], fps=4)
55
  for frame in frames[i]:
56
  writer.append_data(np.array(frame))