Update app.py

app.py

@@ -175,6 +175,7 @@ class AnimateController:
         base_model_dropdown,
         motion_module_dropdown,
         prompt_textbox,
+        init_image,
         negative_prompt_textbox,
         width_slider,
         height_slider,
@@ -202,6 +203,7 @@ class AnimateController:
         
         sample = pipeline(
             prompt_textbox,
+            init_image          = init_image,
             negative_prompt     = negative_prompt_textbox,
             num_inference_steps = 25,
             guidance_scale      = 8.,
@@ -260,6 +262,8 @@ def ui():
                 base_model_dropdown.change(fn=controller.update_base_model,       inputs=[base_model_dropdown],    outputs=[base_model_dropdown])
                 motion_module_dropdown.change(fn=controller.update_motion_module, inputs=[motion_module_dropdown], outputs=[motion_module_dropdown])
 
+                init_image = gr.Image(label="Init Image", source="upload", type="filepath")
+                
                 prompt_textbox          = gr.Textbox( label="Prompt",          lines=3 )
                 negative_prompt_textbox = gr.Textbox( label="Negative Prompt", lines=3, value="worst quality, low quality, nsfw, logo")
 
@@ -278,12 +282,12 @@ def ui():
                 result_video = gr.Video( label="Generated Animation", interactive=False )
                 json_config  = gr.Json( label="Config", value=None )
 
-            inputs  = [base_model_dropdown, motion_module_dropdown, prompt_textbox, negative_prompt_textbox, width_slider, height_slider, seed_textbox]
+            inputs  = [base_model_dropdown, motion_module_dropdown, prompt_textbox, init_image, negative_prompt_textbox, width_slider, height_slider, seed_textbox]
             outputs = [result_video, json_config]
             
             generate_button.click( fn=controller.animate, inputs=inputs, outputs=outputs )
                 
-        gr.Examples( fn=controller.animate, examples=examples, inputs=inputs, outputs=outputs, cache_examples=True )
+        #gr.Examples( fn=controller.animate, examples=examples, inputs=inputs, outputs=outputs, cache_examples=True )
         
     return demo