Update app.py

app.py

@@ -3,6 +3,7 @@ import os
 import shutil
 import yaml
 import tempfile
+import cv2
 import huggingface_hub
 import subprocess
 import threading
@@ -18,22 +19,19 @@ huggingface_hub.login(token=HF_TKN)
 huggingface_hub.hf_hub_download(
     repo_id='yzd-v/DWPose',
     filename='yolox_l.onnx',
-    local_dir='./models/DWPose',
-    local_dir_use_symlinks=False,
+    local_dir='./models/DWPose'
 )
 
 huggingface_hub.hf_hub_download(
     repo_id='yzd-v/DWPose',
     filename='dw-ll_ucoco_384.onnx',
-    local_dir='./models/DWPose',
-    local_dir_use_symlinks=False,
+    local_dir='./models/DWPose'
 )
 
 huggingface_hub.hf_hub_download(
     repo_id='ixaac/MimicMotion',
-    filename='MimicMotion_1-1.pth',
-    local_dir='./models',
-    local_dir_use_symlinks=False,
+    filename='MimicMotion_1.pth',
+    local_dir='./models'
 )
 
 def print_directory_contents(path):
@@ -78,6 +76,21 @@ def check_for_mp4_in_outputs():
     else:
         return None
 
+def get_video_fps(video_path):
+    # Open the video file
+    video_capture = cv2.VideoCapture(video_path)
+    
+    if not video_capture.isOpened():
+        raise ValueError("Error opening video file")
+
+    # Get the FPS value
+    fps = video_capture.get(cv2.CAP_PROP_FPS)
+    
+    # Release the video capture object
+    video_capture.release()
+    
+    return fps
+
 def infer(ref_image_in, ref_video_in):
     # check if 'outputs' dir exists and empty it if necessary
     check_outputs_folder('./outputs')
@@ -89,20 +102,20 @@ def infer(ref_image_in, ref_video_in):
         # Define the values for the variables
         ref_video_path = ref_video_in
         ref_image_path = ref_image_in
-        num_frames = 72
+        num_frames = 16
         resolution = 576
         frames_overlap = 6
         num_inference_steps = 25
         noise_aug_strength = 0
         guidance_scale = 2.0
         sample_stride = 2
-        fps = 15
+        fps = get_video_fps(ref_video_in)
         seed = 42
     
         # Create the data structure
         data = {
             'base_model_path': 'stabilityai/stable-video-diffusion-img2vid-xt-1-1',
-            'ckpt_path': 'models/MimicMotion_1-1.pth',
+            'ckpt_path': 'models/MimicMotion_1.pth',
             'test_case': [
                 {
                     'ref_video_path': ref_video_path,