set output file name and duration to 450

musepose_inference.py

@@ -50,6 +50,7 @@ class MusePoseInference:
         if not os.path.exists(self.output_dir):
             os.makedirs(self.output_dir)
 
+    @spaces.GPU(duration=450)
     def infer_musepose(
         self,
         ref_image_path: str,
@@ -82,11 +83,9 @@ class MusePoseInference:
         print(f"FPS: {fps}")
         print(f"Skip: {skip}")
 
-        image_file_name = os.path.splitext(os.path.basename(ref_image_path))[0]
-        pose_video_file_name = os.path.splitext(os.path.basename(pose_video_path))[0]
-        output_file_name = f"img_{image_file_name}_pose_{pose_video_file_name}"
-        output_path = os.path.abspath(os.path.join(self.output_dir, f'{output_file_name}.mp4'))
-        output_path_demo = os.path.abspath(os.path.join(self.output_dir, f'{output_file_name}_demo.mp4'))
+        output_filename = f"output_temp"
+        output_path = os.path.abspath(os.path.join(self.output_dir, f'{output_filename}.mp4'))
+        output_path_demo = os.path.abspath(os.path.join(self.output_dir, f'{output_filename}_demo.mp4'))
 
         if weight_dtype == "fp16":
             weight_dtype = torch.float16

pose_align.py

@@ -39,7 +39,7 @@ class PoseAlignmentInference:
         if not os.path.exists(self.output_dir):
             os.makedirs(self.output_dir)
 
-    @spaces.GPU(duration=120)
+    @spaces.GPU(duration=450)
     def align_pose(
         self,
         vidfn: str,
@@ -50,10 +50,9 @@ class PoseAlignmentInference:
         max_frame: int,
     ):
         download_models(model_dir=self.model_dir)
-        img_file_name = os.path.splitext(os.path.basename(imgfn_refer))[0]
-        vid_file_name = os.path.splitext(os.path.basename(vidfn))[0]
-        outfn=os.path.abspath(os.path.join(self.output_dir, f'img_{img_file_name}_vid_{vid_file_name}_demo.mp4'))
-        outfn_align_pose_video=os.path.abspath(os.path.join(self.output_dir, f'img_{img_file_name}_vid_{vid_file_name}.mp4'))
+        output_filename = "pose_temp"
+        outfn=os.path.abspath(os.path.join(self.output_dir, f'{output_filename}_demo.mp4'))
+        outfn_align_pose_video=os.path.abspath(os.path.join(self.output_dir, f'{output_filename}.mp4'))
 
         video = cv2.VideoCapture(vidfn)
         width= video.get(cv2.CAP_PROP_FRAME_WIDTH)