File size: 2,306 Bytes
1f4dc2f
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
import gradio as gr 

def create_demo(get_video):
    block = gr.Blocks(css='style.css').queue()
    with block:
        with gr.Group():
            with gr.Box():
                with gr.Row(elem_id='prompt-container').style(equal_height=True):
                    prompt = gr.Text(
                        label='Prompt',
                        show_label=False,
                        max_lines=1,
                        placeholder='Enter your prompt',
                        elem_id='prompt-text-input').style(container=False)
                    run_button = gr.Button('Generate video').style(
                        full_width=False)
            result = gr.Video(label='Result', show_label=False, elem_id='gallery')
            with gr.Accordion('Advanced options', open=False):
                seed = gr.Slider(
                    label='Seed',
                    minimum=-1,
                    maximum=1000000,
                    step=1,
                    value=-1,
                    info='If set to -1, a different seed will be used each time.')
            #     num_frames = gr.Slider(
            #         label='Number of frames',
            #         minimum=16,
            #         maximum=MAX_NUM_FRAMES,
            #         step=1,
            #         value=16,
            #         info=
            #         'Note that the content of the video also changes when you change the number of frames.'
            #     )
            #     num_inference_steps = gr.Slider(label='Number of inference steps',
            #                                     minimum=10,
            #                                     maximum=50,
            #                                     step=1,
            #                                     value=25)
    
        inputs = [
            prompt,
            seed,
            # num_frames,
            # num_inference_steps,
        ]
        gr.Examples(examples=[["Astronaut riding a horse", 10431]],
                    inputs=inputs,
                    outputs=result,
                    fn=get_video,
                    cache_examples=True)
    
        prompt.submit(fn=get_video, inputs=inputs, outputs=result)
        run_button.click(fn=get_video, inputs=inputs, outputs=result)
        
        return block