Update app.py

app.py

@@ -48,17 +48,15 @@ def get_prompt(file:str):
     with open(file,'r') as f:
         a=f.readlines()
     return a #a[0]:positive prompt, a[1] negative prompt
+def unwarp_model(state_dict):
+    new_state_dict = {}
+    for key in state_dict:
+        new_state_dict[key.split('module.')[1]] = state_dict[key]
+    return new_state_dict
 
 def init_pipe():
-    def unwarp_model(state_dict):
-        new_state_dict = {}
-        for key in state_dict:
-            new_state_dict[key.split('module.')[1]] = state_dict[key]
-        return new_state_dict
 
     i2v=True
-    root_path="./"
-    training_steps=0
 
     if i2v:
         key = "i2v"
@@ -179,7 +177,7 @@ def inference(source_images,
 
     return video
 
-def process_video(video_file, image_file, positive_prompt, negative_prompt, guidance, random_seed, choice, progress=None)->str:#gr.Progress(track_tqdm=True)
+def process_video(video_file, image_file, positive_prompt, negative_prompt, guidance, random_seed, choice, progress=gr.Progress(track_tqdm=True))->str:
     if choice==33:
         video_shard=1
     elif choice==65:
@@ -210,7 +208,7 @@ def process_video(video_file, image_file, positive_prompt, negative_prompt, guid
     video:List[PIL.Image.Image]=[]
 
     
-    for i in tqdm(range(video_shard)):
+    for i in progress.tqdm(range(video_shard)):
         if i>0: #first frame guidence
             first_frame=transforms.ToTensor()(video[-1])
             first_frame = first_frame*255.0
@@ -281,7 +279,7 @@ with gr.Blocks() as demo:
             cache_examples=False
         )
     
-demo.launch()
+demo.queue().launch()
 """
 import gradio as gr
 import spaces