Update app.py

app.py

@@ -13,7 +13,7 @@ import spaces
 import torch
 from diffusers import DiffusionPipeline, DPMSolverMultistepScheduler
 
-DESCRIPTION = '# zeroscope v2'
+DESCRIPTION = 'This space is an API service meant to be used by VideoChain and VideoQuest.\nWant to use this space for yourself? Please use the original code: [https://huggingface.co/spaces/hysts/zeroscope-v2](https://huggingface.co/spaces/hysts/zeroscope-v2)'
 if not torch.cuda.is_available():
     DESCRIPTION += '\n<p>Running on CPU 🥶 This demo does not work on CPU.</p>'
 
@@ -21,6 +21,7 @@ MAX_NUM_FRAMES = int(os.getenv('MAX_NUM_FRAMES', '200'))
 DEFAULT_NUM_FRAMES = min(MAX_NUM_FRAMES,
                          int(os.getenv('DEFAULT_NUM_FRAMES', '24')))
 MAX_SEED = np.iinfo(np.int32).max
+SECRET_TOKEN = os.getenv('SECRET_TOKEN', 'default_secret')
 
 if torch.cuda.is_available():
     pipe = DiffusionPipeline.from_pretrained('cerspense/zeroscope_v2_576w',
@@ -49,7 +50,12 @@ def to_video(frames: list[np.ndarray], fps: int) -> str:
 
 @spaces.GPU
 def generate(prompt: str, seed: int, num_frames: int,
-             num_inference_steps: int) -> str:
+             num_inference_steps: int,
+             secret_token: str = '') -> str:
+    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.')
+        
     generator = torch.Generator().manual_seed(seed)
     frames = pipe(prompt,
                   num_inference_steps=num_inference_steps,
@@ -62,9 +68,11 @@ def generate(prompt: str, seed: int, num_frames: int,
 
 with gr.Blocks(css='style.css') as demo:
     gr.Markdown(DESCRIPTION)
-    gr.DuplicateButton(value='Duplicate Space for private use',
-                       elem_id='duplicate-button',
-                       visible=os.getenv('SHOW_DUPLICATE_BUTTON') == '1')
+    secret_token = gr.Text(
+        label='Secret Token',
+        max_lines=1,
+        placeholder='Enter your secret token',
+    )
     with gr.Box():
         with gr.Row():
             prompt = gr.Text(label='Prompt',
@@ -101,6 +109,7 @@ with gr.Blocks(css='style.css') as demo:
         seed,
         num_frames,
         num_inference_steps,
+        secret_token,
     ]
 
     prompt.submit(