Update app.py

app.py

@@ -92,12 +92,9 @@ if torch.cuda.is_available():
     smoother = RIFESmoother.from_model_manager(model_manager2)
 
 
-def change_media(selected, media):
-    if selected == "Diffutoon":
-        media.update(gr.Video(label='Upload Video', height=600, scale=2))
-    elif selected == "ExVideo":
-        media.update(gr.Image(label='Upload Image', height=600, scale=2, image_mode="RGB", type="filepath"))
-    return media
+def change_media(selected):
+    if selected == "ExVideo":
+       return gr.update(gr.Image(label='Upload Image', height=600, scale=2, image_mode="RGB", type="filepath"))
 
 
 
@@ -253,7 +250,7 @@ with gr.Blocks(css=CSS, js=JS, theme="soft") as demo:
         #stop_btn = gr.Button(value="Stop", variant="stop")
         clear_btn = gr.ClearButton([media, seed, video])
 
-    selected.change(change_media, inputs=[selected, media], outputs=media)
+    selected.change(change_media, inputs=[selected], outputs=[media])
     submit_event = submit_btn.click(fn=generate, inputs=[media, selected, seed, num_inference_steps, animatediff_batch_size, animatediff_stride, motion_bucket_id, fps_id, num_frames, prompt], outputs=[video, seed], api_name="video")
     #stop_btn.click(fn=None, inputs=None, outputs=None, cancels=[submit_event])