Update app.py

app.py

@@ -27,7 +27,7 @@ from safetensors.torch import load_file, save_file
 from diffusers import DiffusionPipeline, AnimateDiffPipeline, MotionAdapter, EulerDiscreteScheduler, DDIMScheduler, StableDiffusionXLPipeline, UNet2DConditionModel, AutoencoderKL, UNet3DConditionModel
 #import jax
 #import jax.numpy as jnp
-from numba import cuda, njit as cpu, void, int64 as int, float64 as float, boolean as bool
+from numba import cuda, njit as cpu, void, int64 as int, float64 as float, boolean as bool, unit8 as rgb
 from numba.cuda import jit as gpu, grid
 from numba.types import unicode_type as string
 from PIL.Image import fromarray as array2image
@@ -170,7 +170,7 @@ def generate_random_string(length):
     characters = str(ascii_letters + digits)
     return ''.join(random.choice(characters) for _ in range(length))
     
-@gpu(void())
+@gpu(void( rgb[:], string[:], string[:], string[:] ))
 def calc(img,p1,p2,motion):
     global out_pipe
     global last_motion