Spaces:
Running
Running
update to use gr Examples
Browse files
app.py
CHANGED
@@ -18,8 +18,6 @@ NOTES = 'This app is adapted from <a href="https://github.com/hysts/CogVideo_dem
|
|
18 |
FOOTER = '<img id="visitor-badge" alt="visitor badge" src="https://visitor-badge.glitch.me/badge?page_id=THUDM.CogVideo" />'
|
19 |
|
20 |
|
21 |
-
def set_example_text(example: list) -> dict:
|
22 |
-
return gr.Textbox.update(value=example[0])
|
23 |
|
24 |
|
25 |
def main():
|
@@ -48,7 +46,11 @@ def main():
|
|
48 |
|
49 |
with open('samples.txt') as f:
|
50 |
samples = [[line.strip()] for line in f.readlines()]
|
51 |
-
examples = gr.
|
|
|
|
|
|
|
|
|
52 |
|
53 |
with gr.Column():
|
54 |
with gr.Group():
|
@@ -74,12 +76,8 @@ def main():
|
|
74 |
result_video,
|
75 |
result_gallery,
|
76 |
])
|
77 |
-
examples.click(fn=set_example_text,
|
78 |
-
inputs=examples,
|
79 |
-
outputs=examples.components,
|
80 |
-
queue=False)
|
81 |
|
82 |
-
demo.launch(
|
83 |
|
84 |
|
85 |
if __name__ == '__main__':
|
|
|
18 |
FOOTER = '<img id="visitor-badge" alt="visitor badge" src="https://visitor-badge.glitch.me/badge?page_id=THUDM.CogVideo" />'
|
19 |
|
20 |
|
|
|
|
|
21 |
|
22 |
|
23 |
def main():
|
|
|
46 |
|
47 |
with open('samples.txt') as f:
|
48 |
samples = [[line.strip()] for line in f.readlines()]
|
49 |
+
examples = gr.Examples(examples=samples,
|
50 |
+
fn=model.run_with_translation,
|
51 |
+
inputs=[text,translate,seed,only_first_stage],
|
52 |
+
outputs=[translated_text,result_video,result_gallery],
|
53 |
+
cache_examples=True)
|
54 |
|
55 |
with gr.Column():
|
56 |
with gr.Group():
|
|
|
76 |
result_video,
|
77 |
result_gallery,
|
78 |
])
|
|
|
|
|
|
|
|
|
79 |
|
80 |
+
demo.launch()
|
81 |
|
82 |
|
83 |
if __name__ == '__main__':
|