guardiancc commited on
Commit
50ddafd
·
verified ·
1 Parent(s): 5484118

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -7,16 +7,16 @@ import gradio as gr
7
  from inference_util import init_model, infenrece
8
  from attributtes_utils import input_pose, input_emotion, input_blink
9
 
10
- model = init_model()
11
 
12
  @spaces.GPU
13
  def process(input_vid, audio_path, pose_select, emotion_select, blink_select):
 
14
  pose = input_pose(pose_select)
15
  emotion = input_emotion(emotion_select)
16
  blink = input_blink(blink_select)
17
 
18
  print("input_vid: ", input_vid)
19
- result = infenrece(model, os.path.join("./assets/videos/", input_vid), os.path.join("./assets/audios/", audio_path), pose, emotion, blink)
20
  print("result: ", result)
21
 
22
  print("finished !")
 
7
  from inference_util import init_model, infenrece
8
  from attributtes_utils import input_pose, input_emotion, input_blink
9
 
 
10
 
11
  @spaces.GPU
12
  def process(input_vid, audio_path, pose_select, emotion_select, blink_select):
13
+ model = init_model()
14
  pose = input_pose(pose_select)
15
  emotion = input_emotion(emotion_select)
16
  blink = input_blink(blink_select)
17
 
18
  print("input_vid: ", input_vid)
19
+ result = infenrece(model, input_vid, audio_path, pose, emotion, blink)
20
  print("result: ", result)
21
 
22
  print("finished !")