Update app.py

app.py

@@ -93,19 +93,18 @@ if torch.cuda.is_available():
     smoother = RIFESmoother.from_model_manager(model_manager2)
 
 
-def change_media(image_in, video_in, media, selected):
+def change_media(image_in, video_in, selected):
     if selected == "ExVideo":
-        media = image_in
-        return gr.update(visible=True), gr.update(visible=False), media, gr.update(visible=False)
+        return gr.update(visible=True), gr.update(visible=False), gr.update(visible=False)
     elif selected == "Diffutoon":
-        media = video_in
-        return gr.update(visible=False), gr.update(visible=True), media, gr.update(visible=True)
+        return gr.update(visible=False), gr.update(visible=True), gr.update(visible=True)
 
 
 
 @spaces.GPU(duration=120)
 def generate(
-    media,
+    video_in, 
+    image_in,
     selected,
     prompt: str = "best quality",
     seed: Optional[int] = -1,
@@ -117,8 +116,6 @@ def generate(
     num_frames: int = 50,
     output_folder: str = "outputs",
     progress=gr.Progress(track_tqdm=True)):
-
-    print(media)
     
     if seed == -1:
         seed = random.randint(0, MAX_SEED)
@@ -129,8 +126,8 @@ def generate(
     base_count = len(glob(os.path.join(output_folder, "*.mp4")))
     video_path = os.path.join(output_folder, f"{base_count:06d}.mp4")
 
-    if selected == "ExVideo":
-        image = Image.open(media)
+    if selected == "ExVideo" and image_in:
+        image = Image.open(image_in)
         video = pipe(
             input_image=image.resize((512, 512)), 
             num_frames=num_frames, 
@@ -144,9 +141,9 @@ def generate(
             contrast_enhance_scale=1.2
         )
         model_manager.to("cpu")
-    elif selected == "Diffutoon":
+    elif selected == "Diffutoon" and video_in:
         up_video = VideoData(
-            video_file=media,
+            video_file=video_in,
             height=512, width=512)
         input_video = [up_video[i] for i in range(1, 30)]
 
@@ -189,7 +186,6 @@ with gr.Blocks(css=CSS, js=JS, theme="soft") as demo:
     with gr.Row():
         video_in = gr.Video(label='Upload Video', height=600, scale=2)
         image_in = gr.Image(label='Upload Image', height=600, scale=2, image_mode="RGB", type="filepath", visible=False)
-        media = video_in
         video = gr.Video(label="Generated Video", height=600, scale=2)
         with gr.Column(scale=1):
             selected = gr.Radio(
@@ -256,15 +252,18 @@ with gr.Blocks(css=CSS, js=JS, theme="soft") as demo:
     with gr.Row():
         submit_btn = gr.Button(value="Generate")
         #stop_btn = gr.Button(value="Stop", variant="stop")
-        clear_btn = gr.ClearButton([media, seed, video])
+        clear_btn = gr.ClearButton([video_in, image_in,, seed, video])
         
     gr.Examples(
         examples=examples,
+        fn=generate,
         inputs=[video_in, image_in, selected, prompt],
+        outputs=[video, seed],
+        cache_examples="lazy",
         examples_per_page=4,
     )
-    selected.change(change_media, inputs=[image_in, video_in, media, selected], outputs=[image_in, video_in, media, prompt])
-    submit_event = submit_btn.click(fn=generate, inputs=[media, selected, prompt, seed, num_inference_steps, animatediff_batch_size, animatediff_stride, motion_bucket_id, fps_id, num_frames], outputs=[video, seed], api_name="video")
+    selected.change(change_media, inputs=[image_in, video_in, selected], outputs=[image_in, video_in, prompt])
+    submit_event = submit_btn.click(fn=generate, inputs=[video_in, image_in, selected, prompt, seed, num_inference_steps, animatediff_batch_size, animatediff_stride, motion_bucket_id, fps_id, num_frames], outputs=[video, seed], api_name="video")
     #stop_btn.click(fn=None, inputs=None, outputs=None, cancels=[submit_event])
     
 demo.queue().launch()