BestWishYsh commited on
Commit
0057577
·
verified ·
1 Parent(s): 3337fc1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -2
app.py CHANGED
@@ -9,8 +9,10 @@ import gradio as gr
9
  # import spaces
10
  import torch
11
  from huggingface_hub import hf_hub_download, snapshot_download
12
- from models.consisid_utils import prepare_face_models, process_face_embeddings_infer
13
- from models.pipeline_consisid import ConsisIDPipeline
 
 
14
  from moviepy import VideoFileClip
15
  from util.rife_model import load_rife_model, rife_inference_with_latents
16
  from util.utils import load_sd_upscale, save_video, upscale_batch_and_concatenate
 
9
  # import spaces
10
  import torch
11
  from huggingface_hub import hf_hub_download, snapshot_download
12
+ # from models.consisid_utils import prepare_face_models, process_face_embeddings_infer
13
+ # from models.pipeline_consisid import ConsisIDPipeline
14
+ from diffusers import ConsisIDPipeline
15
+ from diffusers.pipelines.consisid.consisid_utils import prepare_face_models, process_face_embeddings_infer
16
  from moviepy import VideoFileClip
17
  from util.rife_model import load_rife_model, rife_inference_with_latents
18
  from util.utils import load_sd_upscale, save_video, upscale_batch_and_concatenate