Zhouyan248 commited on
Commit
19801e5
1 Parent(s): 26555ee

Update base/app.py

Browse files
Files changed (1) hide show
  1. base/app.py +4 -4
base/app.py CHANGED
@@ -8,8 +8,8 @@ import cv2
8
  import pandas as pd
9
  import torchvision
10
  import random
11
- config_path = "/mnt/petrelfs/zhouyan/project/lavie-release/base/configs/sample.yaml"
12
- args = OmegaConf.load("/mnt/petrelfs/zhouyan/project/lavie-release/base/configs/sample.yaml")
13
  device = "cuda" if torch.cuda.is_available() else "cpu"
14
  # ------- get model ---------------
15
  model_t2V = model_t2v_fun(args)
@@ -84,11 +84,11 @@ with gr.Blocks(css='style.css') as demo:
84
  # outputs = [video_out]
85
  with gr.Column():
86
 
87
- prompt = gr.Textbox(value="a teddy bear walking on the street", label="Prompt", placeholder="enter prompt", show_label=True, elem_id="prompt-in", min_width=200, lines=2)
88
 
89
  ddim_steps = gr.Slider(label='Steps', minimum=50, maximum=300, value=50, step=1)
90
  seed_inp = gr.Slider(value=-1,label="seed (for random generation, use -1)",show_label=True,minimum=-1,maximum=2147483647)
91
- cfg = gr.Number(label="guidance_scale",value=7)
92
  # seed_inp = gr.Slider(label="Seed", minimum=0, maximum=2147483647, step=1, value=400, elem_id="seed-in")
93
 
94
  # with gr.Row():
 
8
  import pandas as pd
9
  import torchvision
10
  import random
11
+ config_path = "./base/configs/sample.yaml"
12
+ args = OmegaConf.load("./base/configs/sample.yaml")
13
  device = "cuda" if torch.cuda.is_available() else "cpu"
14
  # ------- get model ---------------
15
  model_t2V = model_t2v_fun(args)
 
84
  # outputs = [video_out]
85
  with gr.Column():
86
 
87
+ prompt = gr.Textbox(value="a corgi walking in the park at sunrise, oil painting style", label="Prompt", placeholder="enter prompt", show_label=True, elem_id="prompt-in", min_width=200, lines=2)
88
 
89
  ddim_steps = gr.Slider(label='Steps', minimum=50, maximum=300, value=50, step=1)
90
  seed_inp = gr.Slider(value=-1,label="seed (for random generation, use -1)",show_label=True,minimum=-1,maximum=2147483647)
91
+ cfg = gr.Number(label="guidance_scale",value=7.5)
92
  # seed_inp = gr.Slider(label="Seed", minimum=0, maximum=2147483647, step=1, value=400, elem_id="seed-in")
93
 
94
  # with gr.Row():