Update app.py

app.py

@@ -20,6 +20,8 @@ from funcs import (
     save_videos
 )
 
+SECRET_TOKEN = os.getenv('SECRET_TOKEN', 'default_secret')
+
 def download_model():
     REPO_ID = 'Doubiiu/DynamiCrafter_1024'
     filename_list = ['model.ckpt']
@@ -31,7 +33,12 @@ def download_model():
             hf_hub_download(repo_id=REPO_ID, filename=filename, local_dir='./checkpoints/dynamicrafter_1024_v1/', force_download=True)
     
 
-def infer(image, prompt, steps=50, cfg_scale=7.5, eta=1.0, fs=3, seed=123):
+def infer(secret_token, image, prompt, steps=50, cfg_scale=7.5, eta=1.0, fs=3, seed=123):
+    if secret_token != SECRET_TOKEN:
+    raise gr.Error(
+        f'Invalid secret token. Please fork the original space if you want to use it for yourself.')
+
+
     resolution = (576, 1024)
     download_model()
     ckpt_path='checkpoints/dynamicrafter_1024_v1/model.ckpt'
@@ -95,18 +102,6 @@ def infer(image, prompt, steps=50, cfg_scale=7.5, eta=1.0, fs=3, seed=123):
     return video_path
 
 
-i2v_examples = [
-    ['prompts/1024/astronaut04.png', 'a man in an astronaut suit playing a guitar', 50, 7.5, 1.0, 6, 123],
-    ['prompts/1024/bloom01.png', 'time-lapse of a blooming flower with leaves and a stem', 50, 7.5, 1.0, 10, 123],
-    ['prompts/1024/girl07.png', 'a beautiful woman with long hair and a dress blowing in the wind', 50, 7.5, 1.0, 10, 123],
-    ['prompts/1024/pour_bear.png', 'pouring beer into a glass of ice and beer', 50, 7.5, 1.0, 10, 123],
-    ['prompts/1024/robot01.png', 'a robot is walking through a destroyed city', 50, 7.5, 1.0, 10, 123],
-    ['prompts/1024/firework03.png', 'fireworks display', 50, 7.5, 1.0, 10, 123],
-]
-
-
-
-
 css = """#input_img {max-width: 1024px !important} #output_vid {max-width: 1024px; max-height: 576px}"""
 
 with gr.Blocks(analytics_enabled=False, css=css) as dynamicrafter_iface:
@@ -126,30 +121,20 @@ with gr.Blocks(analytics_enabled=False, css=css) as dynamicrafter_iface:
     with gr.Tab(label='ImageAnimation_576x1024'):
         with gr.Column():
             with gr.Row():
-                with gr.Column():
-                    with gr.Row():
-                        i2v_input_image = gr.Image(label="Input Image",elem_id="input_img")
-                    with gr.Row():
-                        i2v_input_text = gr.Text(label='Prompts')
-                    with gr.Row():
-                        i2v_seed = gr.Slider(label='Random Seed', minimum=0, maximum=10000, step=1, value=123)
-                        i2v_eta = gr.Slider(minimum=0.0, maximum=1.0, step=0.1, label='ETA', value=1.0, elem_id="i2v_eta")
-                        i2v_cfg_scale = gr.Slider(minimum=1.0, maximum=15.0, step=0.5, label='CFG Scale', value=7.5, elem_id="i2v_cfg_scale")
-                    with gr.Row():
-                        i2v_steps = gr.Slider(minimum=1, maximum=60, step=1, elem_id="i2v_steps", label="Sampling steps", value=50)
-                        i2v_motion = gr.Slider(minimum=5, maximum=20, step=1, elem_id="i2v_motion", label="FPS", value=10)
-                    i2v_end_btn = gr.Button("Generate")
-                # with gr.Tab(label='Result'):
-                with gr.Row():
-                    i2v_output_video = gr.Video(label="Generated Video",elem_id="output_vid",autoplay=True,show_share_button=True)
-
-            gr.Examples(examples=i2v_examples,
-                        inputs=[i2v_input_image, i2v_input_text, i2v_steps, i2v_cfg_scale, i2v_eta, i2v_motion, i2v_seed],
-                        outputs=[i2v_output_video],
-                        fn = infer,
-                        cache_examples=True,
-            )
-        i2v_end_btn.click(inputs=[i2v_input_image, i2v_input_text, i2v_steps, i2v_cfg_scale, i2v_eta, i2v_motion, i2v_seed],
+                secret_token = gr.Text(label='Secret Token', max_lines=1)
+                i2v_input_image = gr.Image(label="Input Image",elem_id="input_img")
+                i2v_input_text = gr.Text(label='Prompts')
+                i2v_seed = gr.Slider(label='Random Seed', minimum=0, maximum=10000, step=1, value=123)
+                i2v_eta = gr.Slider(minimum=0.0, maximum=1.0, step=0.1, label='ETA', value=1.0, elem_id="i2v_eta")
+                i2v_cfg_scale = gr.Slider(minimum=1.0, maximum=15.0, step=0.5, label='CFG Scale', value=7.5, elem_id="i2v_cfg_scale")
+                i2v_steps = gr.Slider(minimum=1, maximum=60, step=1, elem_id="i2v_steps", label="Sampling steps", value=50)
+                i2v_motion = gr.Slider(minimum=5, maximum=20, step=1, elem_id="i2v_motion", label="FPS", value=10)
+                i2v_end_btn = gr.Button("Generate")
+
+            with gr.Row():
+                i2v_output_video = gr.Video(label="Generated Video",elem_id="output_vid",autoplay=True,show_share_button=True)
+
+        i2v_end_btn.click(inputs=[secret_token, i2v_input_image, i2v_input_text, i2v_steps, i2v_cfg_scale, i2v_eta, i2v_motion, i2v_seed],
                         outputs=[i2v_output_video],
                         fn = infer
         )