Update app.py

app.py

@@ -37,6 +37,7 @@ css = """
 }
 """
 
+SECRET_TOKEN = os.getenv('SECRET_TOKEN', 'default_secret')
 
 class AnimateController:
     def __init__(self):
@@ -156,6 +157,7 @@ class AnimateController:
     @torch.no_grad()
     def animate(
         self,
+        secret_token,
         lora_alpha_slider,
         spatial_lora_slider,
         prompt_textbox,
@@ -168,6 +170,10 @@ class AnimateController:
         cfg_scale_slider,
         seed_textbox
     ):
+        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.')
+
 
         if is_xformers_available():
             self.unet.enable_xformers_memory_efficient_attention()
@@ -247,8 +253,9 @@ def ui():
             2. Increase the sampling step and cfg and set proper negative prompt if you want more fancy videos.
             '''
         )
-        with gr.Column(variant="panel"):
+        with gr.Column():
             with gr.Row():
+                 secret_token = gr.Text(label='Secret Token', max_lines=1)
 
                 base_model_dropdown = gr.Dropdown(
                     label="Select base Dreambooth model (required)",
@@ -331,6 +338,7 @@ def ui():
             generate_button.click(
                 fn=controller.animate,
                 inputs=[
+                    secret_token,
                     lora_alpha_slider,
                     spatial_lora_slider,
                     prompt_textbox,
@@ -346,34 +354,6 @@ def ui():
                 outputs=[result_video]
             )
             
-            examples = [
-                [0.8, 0.8, "a boy is holding a rabbit", "bad quality", "LCM", 8, 512, 16, 512, 1.5, 123],
-                [0.8, 0.8, "1girl smiling", "bad quality", "LCM", 4, 512, 16, 512, 1.5, 1233],
-                [0.8, 0.8, "1girl,face,white background,", "bad quality", "LCM", 6, 512, 16, 512, 1.5, 1234],
-                [0.8, 0.8, "clouds in the sky, best quality", "bad quality", "LCM", 4, 512, 16, 512, 1.5, 1234],
-                
-                
-            ]
-            gr.Examples(
-                examples = examples,
-                inputs=[
-                    lora_alpha_slider,
-                    spatial_lora_slider,
-                    prompt_textbox,
-                    negative_prompt_textbox,
-                    sampler_dropdown,
-                    sample_step_slider,
-                    width_slider,
-                    length_slider,
-                    height_slider,
-                    cfg_scale_slider,
-                    seed_textbox,
-                ],
-                outputs=[result_video],
-                fn=controller.animate,
-                cache_examples=True,
-            )
-
     return demo
 
 
@@ -382,5 +362,5 @@ if __name__ == "__main__":
     demo = ui()
     # gr.close_all()
     # restart
-    demo.queue(api_open=False)
+    demo.queue(api_open=True)
     demo.launch()