Update app.py

app.py

@@ -93,11 +93,13 @@ if torch.cuda.is_available():
     smoother = RIFESmoother.from_model_manager(model_manager2)
 
 
-def change_media(image_in, video_in, selected):
+def change_media(image_in, video_in, media, selected):
     if selected == "ExVideo":
-        return gr.update(visible=True), gr.update(visible=False), image_in, gr.update(visible=False)
+        media = image_in
+        return gr.update(visible=True), gr.update(visible=False), media, gr.update(visible=False)
     elif selected == "Diffutoon":
-        return gr.update(visible=False), gr.update(visible=True), video_in, gr.update(visible=True)
+        media = video_in
+        return gr.update(visible=False), gr.update(visible=True), media, gr.update(visible=True)
 
 
 
@@ -261,7 +263,7 @@ with gr.Blocks(css=CSS, js=JS, theme="soft") as demo:
         inputs=[video_in, image_in, selected, prompt],
         examples_per_page=4,
     )
-    selected.change(change_media, inputs=[image_in, video_in, selected], outputs=[image_in, video_in, media, prompt])
+    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")
     #stop_btn.click(fn=None, inputs=None, outputs=None, cancels=[submit_event])