Spaces:
Runtime error
Runtime error
Zhouyan248
commited on
Commit
β’
b97495d
1
Parent(s):
f5cd544
Update app.py
Browse files
app.py
CHANGED
@@ -63,7 +63,6 @@ def infer(prompt, image_inp, seed_inp, ddim_steps):
|
|
63 |
video_input = video_input.to(device).unsqueeze(0)
|
64 |
mask = mask_generation_before(args.mask_type, video_input.shape, video_input.dtype, device)
|
65 |
masked_video = video_input * (mask == 0)
|
66 |
-
prompt = "tilt up, high quality, stable "
|
67 |
prompt = prompt + args.additional_prompt
|
68 |
video_clip = auto_inpainting(args, video_input, masked_video, mask, prompt, vae, text_encoder, diffusion, model, device,)
|
69 |
video_ = ((video_clip * 0.5 + 0.5) * 255).add_(0.5).clamp_(0, 255).to(dtype=torch.uint8).cpu().permute(0, 2, 3, 1)
|
|
|
63 |
video_input = video_input.to(device).unsqueeze(0)
|
64 |
mask = mask_generation_before(args.mask_type, video_input.shape, video_input.dtype, device)
|
65 |
masked_video = video_input * (mask == 0)
|
|
|
66 |
prompt = prompt + args.additional_prompt
|
67 |
video_clip = auto_inpainting(args, video_input, masked_video, mask, prompt, vae, text_encoder, diffusion, model, device,)
|
68 |
video_ = ((video_clip * 0.5 + 0.5) * 255).add_(0.5).clamp_(0, 255).to(dtype=torch.uint8).cpu().permute(0, 2, 3, 1)
|