Update app.py

app.py

@@ -28,7 +28,7 @@ from diffusers import DiffusionPipeline, AnimateDiffPipeline, MotionAdapter, Eul
 #import jax
 #import jax.numpy as jnp
 from numba import cuda, njit as cpu, void, int64 as int, float64 as float, boolean as bool, uint8 as rgb
-from numba.cuda import jit as gpu, grid
+from numba.cuda import jit as gpu, grid, as_cuda_array as tensor2array
 from numba.types import unicode_type as string
 from PIL.Image import fromarray as array2image
 import numpy as np
@@ -119,6 +119,23 @@ function custom(){
 }
 """
 
+# torch pipe
+
+pipe = AnimateDiffPipeline.from_pretrained(base, vae=vae, motion_adapter=adapter, torch_dtype=dtype).to(device)
+pipe.scheduler = DDIMScheduler(
+    clip_sample=False,
+    beta_start=0.00085,
+    beta_end=0.012,
+    beta_schedule="linear",
+    timestep_spacing="trailing",
+    steps_offset=1
+)
+pipe.unet.load_state_dict(load_file(hf_hub_download(repo, ckpt), device=device), strict=False)
+pipe.load_ip_adapter("h94/IP-Adapter", subfolder="models", weight_name="ip-adapter_sd15.bin")
+pipe.enable_vae_slicing()
+pipe.enable_free_init(method="butterworth", use_fast_sampling=fast)
+
+
 # functionality
 
 def run(cmd):
@@ -174,6 +191,7 @@ def generate_random_string(length):
 def calc(img,p1,p2,motion):
     global out_pipe
     global last_motion
+    global pipe
 
     x = grid(1)
 
@@ -295,23 +313,6 @@ def ui():
                 result.append(gr.Image(interactive=False,elem_classes="image-container", label="Result", show_label=False, type='filepath', show_share_button=False))
         demo.queue().launch()
 
-@gpu(void())
-def pre():
-    global pipe
-    pipe = AnimateDiffPipeline.from_pretrained(base, vae=vae, motion_adapter=adapter, torch_dtype=dtype).to(device)
-    pipe.scheduler = DDIMScheduler(
-        clip_sample=False,
-        beta_start=0.00085,
-        beta_end=0.012,
-        beta_schedule="linear",
-        timestep_spacing="trailing",
-        steps_offset=1
-    )
-    pipe.unet.load_state_dict(load_file(hf_hub_download(repo, ckpt), device=device), strict=False)
-    pipe.load_ip_adapter("h94/IP-Adapter", subfolder="models", weight_name="ip-adapter_sd15.bin")
-    pipe.enable_vae_slicing()
-    pipe.enable_free_init(method="butterworth", use_fast_sampling=fast)
-
 @gpu(void())
 def events():
     gr.on(
@@ -327,7 +328,7 @@ def events():
 
 def entry():
     os.chdir(os.path.abspath(os.path.dirname(__file__)))
-    pre[1,32]()
+    
     ui()
     events[1,32]()