Update app.py

app.py

@@ -24,7 +24,7 @@ huggingface_hub.hf_hub_download(
 
 huggingface_hub.hf_hub_download(
     repo_id='ixaac/MimicMotion',
-    filename='MimicMotion_1-1.pth',
+    filename='MimicMotion_1.pth',
     local_dir='./models',
     local_dir_use_symlinks=False,
 )
@@ -52,14 +52,14 @@ def infer(ref_video_in, ref_image_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 = 12
         seed = 42
     
         # Create the data structure
@@ -93,12 +93,24 @@ def infer(ref_video_in, ref_image_in):
         print("YAML file 'config.yaml' created successfully in", file_path)
 
         # Execute the inference command
-        command = ['python', 'inference.py', '--inference_config', file_path]
-        result = subprocess.run(command, capture_output=True, text=True)
-    
-        # Print the command output
-        print("Command output:", result.stdout)
-        print("Command errors:", result.stderr)
+        process = subprocess.Popen(command, stdout=subprocess.PIPE, stderr=subprocess.PIPE, text=True)
+        # Print logs in real-time
+        while True:
+            output = process.stdout.readline()
+            if output == '' and process.poll() is not None:
+                break
+            if output:
+                print(output.strip())
+        
+        # Print any remaining output
+        for output in process.stdout:
+            print(output.strip())
+        for error in process.stderr:
+            print(error.strip())
+        
+        # Wait for the process to complete and get the return code
+        return_code = process.wait()
+        print("Inference script finished with return code:", return_code)
     
     
     return "done"