saicharan1234 commited on
Commit
69dd2a2
1 Parent(s): 7b988f1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -4
app.py CHANGED
@@ -33,7 +33,7 @@ def run_inference(source_image, driving_audio, pose_weight, face_weight, lip_wei
33
  inference_process(args)
34
  return f'output-{unique_id}.mp4'
35
 
36
- with gr.Blocks() as demo:
37
  gr.Markdown(
38
  """
39
  # Talking Head Generation
@@ -46,6 +46,10 @@ with gr.Blocks() as demo:
46
  avatar_face = gr.Image(type="filepath", label="Face", elem_id="face-input")
47
  driving_audio = gr.Audio(type="filepath", label="Driving Audio", elem_id="audio-input")
48
 
 
 
 
 
49
  with gr.Accordion("Advanced Settings", open=False):
50
  pose_weight = gr.Slider(minimum=0.0, value=1.5, label="Pose Weight")
51
  face_weight = gr.Slider(minimum=0.0, value=1.0, label="Face Weight")
@@ -53,9 +57,6 @@ with gr.Blocks() as demo:
53
  face_expand_ratio = gr.Slider(minimum=0.0, value=1.2, label="Face Expand Ratio")
54
 
55
  generate = gr.Button("Generate", elem_id="generate-button")
56
-
57
- with gr.Column():
58
- output_video = gr.Video(label="Your Talking Head", elem_id="output-video")
59
 
60
  generate.click(
61
  fn=run_inference,
 
33
  inference_process(args)
34
  return f'output-{unique_id}.mp4'
35
 
36
+ with gr.Blocks(theme='freddyaboulton/dracula_revamped@0.3.8' ) as demo:
37
  gr.Markdown(
38
  """
39
  # Talking Head Generation
 
46
  avatar_face = gr.Image(type="filepath", label="Face", elem_id="face-input")
47
  driving_audio = gr.Audio(type="filepath", label="Driving Audio", elem_id="audio-input")
48
 
49
+
50
+
51
+ with gr.Column():
52
+ output_video = gr.Video(label="Your Talking Head", elem_id="output-video")
53
  with gr.Accordion("Advanced Settings", open=False):
54
  pose_weight = gr.Slider(minimum=0.0, value=1.5, label="Pose Weight")
55
  face_weight = gr.Slider(minimum=0.0, value=1.0, label="Face Weight")
 
57
  face_expand_ratio = gr.Slider(minimum=0.0, value=1.2, label="Face Expand Ratio")
58
 
59
  generate = gr.Button("Generate", elem_id="generate-button")
 
 
 
60
 
61
  generate.click(
62
  fn=run_inference,