Update app.py

app.py

@@ -12,11 +12,15 @@ from realesrgan.archs.srvgg_arch import SRVGGNetCompact
 
 last_file = None
 img_mode = "RGBA"
+SECRET_TOKEN = os.getenv('SECRET_TOKEN', 'default_secret')
 
-
-def realesrgan(img, model_name, denoise_strength, face_enhance, outscale):
+def realesrgan(secret_token, img, model_name, denoise_strength, face_enhance, outscale):
     """Real-ESRGAN function to restore (and upscale) images.
     """
+    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 not img:
         return
 
@@ -176,7 +180,11 @@ def main():
         gr.Markdown(
             """This Space is a fork of "Real-ESRGAN-Demo", so if you want to use it please refer to [havas79/Real-ESRGAN_Demo](https://huggingface.co/spaces/havas79/Real-ESRGAN_Demo), thank you!"""
         )
-
+        secret_token = gr.Text(
+            label='Secret Token',
+            max_lines=1,
+            placeholder='Enter your secret token',
+        )
         with gr.Accordion("Options/Parameters"):
             with gr.Row():
                 model_name = gr.Dropdown(label="Real-ESRGAN inference model to be used",
@@ -201,7 +209,7 @@ def main():
         # Event listeners:
         input_image.change(fn=image_properties, inputs=input_image, outputs=input_image_properties)
         restore_btn.click(fn=realesrgan,
-                          inputs=[input_image, model_name, denoise_strength, face_enhance, outscale],
+                          inputs=[secret_token, input_image, model_name, denoise_strength, face_enhance, outscale],
                           outputs=output_image,
                           api_name="upscale")