Update app_swin.py

app_swin.py

@@ -272,8 +272,7 @@ def save_video_to_mp4(video_tensor, fps=10):
     video_numpy = video_tensor.permute(0, 2, 3, 1).cpu().numpy()
     # Normalize values to [0, 255] if necessary
     if video_numpy.max() <= 1.0:
-        #video_numpy = (video_numpy / 2 + 0.5).astype(np.uint8)  # Unnormalize
-        video_numpy = (video_numpy * 255).astype(np.uint8)
+        video_numpy = ((video_tensor + 1) / 2 * 255).astype(np.uint8) # Unnormalize
     
     # Create a temporary file to save the video
     temp_file = tempfile.NamedTemporaryFile(delete=False, suffix=".mp4")