Update app.py

app.py

@@ -92,9 +92,11 @@ if torch.cuda.is_available():
     smoother = RIFESmoother.from_model_manager(model_manager2)
 
 
-def change_media(selected):
+def change_media(image_in, video_in, selected):
     if selected == "ExVideo":
-       return gr.update(gr.Image(label='Upload Image', height=600, scale=2, image_mode="RGB", type="filepath"))
+        return gr.update(visible=True), gr.update(visible=False), image_in
+    elif selected == "Diffutoon":
+        return gr.update(visible=True), gr.update(visible=False), video_in
 
 
 
@@ -181,7 +183,9 @@ with gr.Blocks(css=CSS, js=JS, theme="soft") as demo:
     gr.HTML("<h1><center>Exvideo📽️Diffutoon</center></h1>")
     gr.HTML("<p><center>Exvideo and Diffutoon video generation<br><b>Update</b>: first version<br><b>Note</b>: ZeroGPU limited, Set the parameters appropriately.</center></p>")
     with gr.Row():
-        media = gr.Video(label='Upload Video', height=600, scale=2)
+        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(
@@ -250,7 +254,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], outputs=[media])
+    selected.change(change_media, inputs=[image_in, video_in, selected], outputs=[image_in, video_in, 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])