a

app.py

@@ -17,7 +17,7 @@ examples = [['project/cartoon2.jpg','project/video1.mp4'],
 
 title = "DaGAN"
 description = """
-Gradio demo for <b>Depth-Aware Generative Adversarial Network for Talking Head Video Generation</b>, CVPR 2022L. <a href='https://arxiv.org/abs/2203.06605'>[Paper]</a><a href='https://github.com/harlanhong/CVPR2022-DaGAN'>[Github Code]</a>\n 
+Gradio demo for <b>Depth-Aware Generative Adversarial Network for Talking Head Video Generation</b>, CVPR 2022. <a href='https://arxiv.org/abs/2203.06605'>[Paper]</a><a href='https://github.com/harlanhong/CVPR2022-DaGAN'>[Github Code]</a>\n 
 """
 ##With Restormer, you can perform: (1) Image Denoising, (2) Defocus Deblurring, (3)  Motion Deblurring, and (4) Image Deraining. 
 ##To use it, simply upload your own image, or click one of the examples provided below.
@@ -28,6 +28,10 @@ article = "<p style='text-align: center'><a href='https://arxiv.org/abs/2203.066
 def inference(img, video):
     if not os.path.exists('temp'):
       os.system('mkdir temp')
+	 # trim video to 8 seconds
+    cmd = f"ffmpeg -y -ss 00:00:00 -i {video} -to 00:00:08 -c copy video_input.mp4"
+    subprocess.run(cmd.split())
+    video = "video_input.mp4"
     ####  Resize the longer edge of the input image
     # os.system("ffmpeg -y -ss 00:00:00 -i {video} -to 00:00:08 -c copy temp/driving_video.mp4")
     # driving_video = "video_input.mp4"
@@ -38,7 +42,7 @@ gr.Interface(
 		inference,
 		[
 				gr.inputs.Image(type="filepath", label="Source Image"),
-				gr.inputs.Video(type='filepath',label="Driving Video"),
+				gr.inputs.Video(type='mp4',label="Driving Video"),
 		],
 		gr.outputs.Video(type="mp4", label="Output Video"),
 		title=title,