Update app.py

app.py

@@ -138,10 +138,10 @@ def inference(source_images,
     pipe.controlnet_transformer.to(DEVICE)
 
     source_pixel_values = source_images/127.5 - 1.0
-    source_pixel_values = source_pixel_values.to(torch.float16).to("cuda:0")
+    source_pixel_values = source_pixel_values.to(torch.float16).to(DEVICE)
     if target_images is not None:
         target_pixel_values = target_images/127.5 - 1.0
-        target_pixel_values = target_pixel_values.to(torch.float16).to("cuda:0")
+        target_pixel_values = target_pixel_values.to(torch.float16).to(DEVICE)
     bsz,f,h,w,c = source_pixel_values.shape
 
     with torch.no_grad():
@@ -172,7 +172,7 @@ def inference(source_images,
         height = h,
         width = w,
         num_frames = f,
-        num_inference_steps = 50,
+        num_inference_steps = 5,#!!!
         interval = 6,
         guidance_scale = guidance_scale,
         generator = torch.Generator(device=device).manual_seed(random_seed)
@@ -237,14 +237,6 @@ def process_video(video_file, image_file, positive_prompt, negative_prompt, guid
 PIPE=init_pipe()
 
 
-def process_video(video_file, image_file, positive_prompt, negative_prompt, guidance, random_seed, choice, progress=gr.Progress(track_tqdm=True))->str:
-    return "123456"
-
-def get_prompt(file:str):
-    with open(file,'r') as f:
-        a=f.readlines()
-    return a #a[0]:positive prompt, a[1] negative prompt
-
 with gr.Blocks() as demo:
     gr.Markdown(
         """