Update app.py

app.py

@@ -2,9 +2,14 @@ import gradio as gr
 from modelscope.pipelines import pipeline
 from modelscope.outputs import OutputKeys
 
+SECRET_TOKEN = os.getenv('SECRET_TOKEN', 'default_secret')
+
 pipe = pipeline(task='image-to-video', model='damo/Image-to-Video', model_revision='v1.1.0')
 
-def infer (image_in):
+def infer (image_in, secret_token):
+    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.')
+        
     # IMG_PATH: your image path (url or local file)
     IMG_PATH = image_in
     output_video_path = pipe(IMG_PATH, output_video='output.mp4')[OutputKeys.OUTPUT_VIDEO]
@@ -18,6 +23,7 @@ with gr.Blocks() as demo:
         For the human UI, please check out the <a href="https://huggingface.co/spaces/fffiloni/MS-Image2Video">original Space by Sylvain Filoni</a>.
      </p>
     """)
+    secret_token = gr.Text(label='Secret', max_lines=1, placeholder='secret')
     image_in = gr.Image(
         label = "Source Image",
         source = "upload", 
@@ -26,6 +32,6 @@ with gr.Blocks() as demo:
     )
     submit_btn = gr.Button("Submit")
     video_out = gr.Video(label = "Video Result", elem_id = "video-out")
-    submit_btn.click(fn = infer, inputs = [image_in], outputs = [video_out])
+    submit_btn.click(fn = infer, inputs = [image_in, secret_token], outputs = [video_out], api_name ="generate")
 
 demo.queue(max_size=6).launch()