yufengzhu commited on
Commit
66bb4b2
·
1 Parent(s): b1a2d4d
Files changed (1) hide show
  1. app.py +11 -4
app.py CHANGED
@@ -54,7 +54,7 @@ def post(
54
  writer.append_data(np.array(frame))
55
  writer.close()
56
  print('finish')
57
- return translated_text, result_video[0]
58
 
59
  def main():
60
  only_first_stage = True
@@ -87,10 +87,17 @@ def main():
87
 
88
  with gr.Column():
89
  with gr.Group():
90
- translated_text = gr.Textbox(label='Translated Text')
91
  with gr.Tabs():
92
  with gr.TabItem('Output1 (Video)'):
93
- result_video = gr.Video(show_label=False)
 
 
 
 
 
 
 
94
 
95
 
96
  # examples = gr.Examples(
@@ -112,7 +119,7 @@ def main():
112
  only_first_stage,
113
  image_prompt
114
  ],
115
- outputs=[translated_text, result_video])
116
  print(gr.__version__)
117
 
118
  demo.launch()
 
54
  writer.append_data(np.array(frame))
55
  writer.close()
56
  print('finish')
57
+ return translated_text, result_video[0], result_video[1], result_video[2], result_video[3]
58
 
59
  def main():
60
  only_first_stage = True
 
87
 
88
  with gr.Column():
89
  with gr.Group():
90
+ #translated_text = gr.Textbox(label='Translated Text')
91
  with gr.Tabs():
92
  with gr.TabItem('Output1 (Video)'):
93
+ result_video1 = gr.Video(show_label=False)
94
+ with gr.TabItem('Output2 (Video)'):
95
+ result_video2 = gr.Video(show_label=False)
96
+ with gr.TabItem('Output3 (Video)'):
97
+ result_video3 = gr.Video(show_label=False)
98
+ with gr.TabItem('Output4 (Video)'):
99
+ result_video4 = gr.Video(show_label=False)
100
+
101
 
102
 
103
  # examples = gr.Examples(
 
119
  only_first_stage,
120
  image_prompt
121
  ],
122
+ outputs=[translated_text, result_video1, result_video2, result_video3, result_video4])
123
  print(gr.__version__)
124
 
125
  demo.launch()