Update app.py

app.py

@@ -24,6 +24,8 @@ from audiocraft.data.audio import audio_write
 from audiocraft.models import MAGNeT
 
 
+SECRET_TOKEN = os.getenv('SECRET_TOKEN', 'default_secret')
+
 MODEL = None  # Last used model
 SPACE_ID = os.environ.get('SPACE_ID', '')
 MAX_BATCH_SIZE = 12
@@ -131,11 +133,15 @@ def predict_batched(texts, melodies):
     res = _do_predictions(texts, melodies)
     return res
 
-def predict_full(model, model_path, text, temperature, topp,
+def predict_full(secre_token, model, model_path, text, temperature, topp,
                  max_cfg_coef, min_cfg_coef, 
                  decoding_steps1, decoding_steps2, decoding_steps3, decoding_steps4, 
                  span_score,
                  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.')
+
     global INTERRUPTING
     INTERRUPTING = False
     progress(0, desc="Loading model...")
@@ -191,15 +197,13 @@ def predict_full(model, model_path, text, temperature, topp,
 
 def ui_full(launch_kwargs):
     with gr.Blocks() as interface:
-        gr.Markdown(
-            """
-            # MAGNeT
-            This is a temporary demo for [MAGNeT](https://github.com/facebookresearch/audiocraft),
-            [running from <a href="https://github.com/facebookresearch/audiocraft/commit/7dece43a4d186e47e5e1c67983ed10a99f225948" target="_blank">"magnet_xformers_0_0_22_fix" audiocraft branch</a>]<br />
-            A fast text-to-music model, consists of a single, non-autoregressive transformer.
-            presented at: ["Masked Audio Generation using a Single Non-Autoregressive Transformer"] (https://huggingface.co/papers/2401.04577)
-            """
-        )
+        gr.HTML("""
+            <div style="z-index: 100; position: fixed; top: 0px; right: 0px; left: 0px; bottom: 0px; width: 100%; height: 100%; background: white; display: flex; align-items: center; justify-content: center; color: black;">
+            <div style="text-align: center; color: black;">
+            <p style="color: black;">This space is a headless component of the cloud rendering engine used by AiTube.</p>
+            <p style="color: black;">It is not available for public use, but you can use the <a href="https://huggingface.co/spaces/doevent/AnimateLCM-SVD" target="_blank">original space</a>.</p>
+            </div>
+            </div>""")
         with gr.Row():
             with gr.Column():
                 secret_token = gr.Textbox((label="Secret Token")