EvanTHU commited on
Commit
fafa3a5
·
verified ·
1 Parent(s): 0f38a1c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +16 -0
app.py CHANGED
@@ -141,6 +141,22 @@ CSS = """
141
 
142
  @spaces.GPU
143
  def generate_video_from_text(text, opt, pipeline):
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
144
  width = 500
145
  height = 500
146
  texts = [text, text]
 
141
 
142
  @spaces.GPU
143
  def generate_video_from_text(text, opt, pipeline):
144
+ global edit_config
145
+
146
+ gr.Info("Loading Configurations...", duration = 3)
147
+ model = build_models(opt, edit_config=edit_config)
148
+ ckpt_path = pjoin(opt.model_dir, opt.which_ckpt + '.tar')
149
+ niter = load_model_weights(model, ckpt_path, use_ema=not opt.no_ema)
150
+
151
+ pipeline = DiffusePipeline(
152
+ opt = opt,
153
+ model = model,
154
+ diffuser_name = opt.diffuser_name,
155
+ device=opt.device,
156
+ num_inference_steps=opt.num_inference_steps,
157
+ torch_dtype=torch.float16,
158
+ )
159
+
160
  width = 500
161
  height = 500
162
  texts = [text, text]