isort

app.py

@@ -1,28 +1,31 @@
-import spaces
 import datetime
+import os
+import sys
 import uuid
-from PIL import Image
-import numpy as np
-import cv2
-from scipy.interpolate import interp1d, PchipInterpolator
-from packaging import version
 
+import cv2
+import gradio as gr
+import numpy as np
+import spaces
 import torch
 import torchvision
-import gradio as gr
+from diffusers.utils import export_to_gif, export_to_video, load_image
 # from moviepy.editor import *
 from diffusers.utils.import_utils import is_xformers_available
-from diffusers.utils import load_image, export_to_video, export_to_gif
+from packaging import version
+from PIL import Image
+from scipy.interpolate import PchipInterpolator, interp1d
 
-import os
-import sys
 sys.path.insert(0, os.getcwd())
-from models_diffusers.controlnet_svd import ControlNetSVDModel
-from models_diffusers.unet_spatio_temporal_condition import UNetSpatioTemporalConditionModel
-from pipelines.pipeline_stable_video_diffusion_interp_control import StableVideoDiffusionInterpControlPipeline
+import warnings
+
 from gradio_demo.utils_drag import *
+from models_diffusers.controlnet_svd import ControlNetSVDModel
+from models_diffusers.unet_spatio_temporal_condition import \
+    UNetSpatioTemporalConditionModel
+from pipelines.pipeline_stable_video_diffusion_interp_control import \
+    StableVideoDiffusionInterpControlPipeline
 
-import warnings
 print("gr file", gr.__file__)
 
 from huggingface_hub import hf_hub_download, snapshot_download
@@ -416,8 +419,9 @@ class Drag:
             sift_track_update = True
             controlnet_cond_scale = 0.5
 
+            from models_diffusers.sift_match import \
+                interpolate_trajectory as sift_interpolate_trajectory
             from models_diffusers.sift_match import sift_match
-            from models_diffusers.sift_match import interpolate_trajectory as sift_interpolate_trajectory
 
             output_file_sift = os.path.join(args.output_dir,  "sift.png")