Spaces:
Running
on
A10G
Running
on
A10G
Zhouyan248
commited on
Commit
•
19801e5
1
Parent(s):
26555ee
Update base/app.py
Browse files- 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 = "
|
12 |
-
args = OmegaConf.load("
|
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
|
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():
|