Update app.py

app.py

@@ -290,6 +290,7 @@ def generate_video_from_text(
 
 
 def generate_video_from_image(
+    secret_token="",
     image_path,
     prompt="",
     enhance_prompt_toggle=False,
@@ -303,6 +304,9 @@ def generate_video_from_image(
     num_frames=121,
     progress=gr.Progress(),
 ):
+    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.')
 
     print("Height: ", height)
     print("Width: ", width)
@@ -616,6 +620,7 @@ with gr.Blocks(theme=gr.themes.Soft()) as iface:
     img2vid_generate.click(
         fn=generate_video_from_image,
         inputs=[
+            secret_token,
             img2vid_image,
             img2vid_prompt,
             img2vid_enhance_toggle,