public demo warning

app.py

@@ -2,12 +2,8 @@ import gradio as gr
 import torch
 from finetuning import FineTunedModel
 from StableDiffuser import StableDiffuser
-from tqdm import tqdm
 from train import train
-
-#import os
-#os.system("nvidia-smi topo -m")
-
+import os
 model_map = {'Van Gogh' : 'models/vangogh.pt', 
              'Pablo Picasso': 'models/pablopicasso.pt',
              'Car' : 'models/car.pt',
@@ -19,6 +15,13 @@ model_map = {'Van Gogh' : 'models/vangogh.pt',
              'Kelly McKernan': 'models/kellymckernan.pt',
              'Rembrandt': 'models/rembrandt.pt' }
 
+ORIGINAL_SPACE_ID = 'baulab/Erasing-Concepts-In-Diffusion'
+SPACE_ID = os.getenv('SPACE_ID')
+
+SHARED_UI_WARNING = f'''## Attention - Training using the ESD-u method does not work in this shard UI. You can either duplicate and use it with a gpu with at least 40GB, or clone this repository to run on your own machine.
+<center><a class="duplicate-button" style="display:inline-block" target="_blank" href="https://huggingface.co/spaces/{SPACE_ID}?duplicate=true"><img style="margin-top:0;margin-bottom:0" src="https://img.shields.io/badge/-Duplicate%20Space-blue?labelColor=white&style=flat&logo=data:image/png;base64,iVBORw0KGgoAAAANSUhEUgAAABAAAAAQCAYAAAAf8/9hAAAAAXNSR0IArs4c6QAAAP5JREFUOE+lk7FqAkEURY+ltunEgFXS2sZGIbXfEPdLlnxJyDdYB62sbbUKpLbVNhyYFzbrrA74YJlh9r079973psed0cvUD4A+4HoCjsA85X0Dfn/RBLBgBDxnQPfAEJgBY+A9gALA4tcbamSzS4xq4FOQAJgCDwV2CPKV8tZAJcAjMMkUe1vX+U+SMhfAJEHasQIWmXNN3abzDwHUrgcRGmYcgKe0bxrblHEB4E/pndMazNpSZGcsZdBlYJcEL9Afo75molJyM2FxmPgmgPqlWNLGfwZGG6UiyEvLzHYDmoPkDDiNm9JR9uboiONcBXrpY1qmgs21x1QwyZcpvxt9NS09PlsPAAAAAElFTkSuQmCC&logoWidth=14" alt="Duplicate Space"></a></center>
+'''
+
 
 class Demo:
 
@@ -36,6 +39,12 @@ class Demo:
 
     def layout(self):
 
+        with gr.Row():
+
+            if SPACE_ID == ORIGINAL_SPACE_ID:
+
+                self.warning = gr.Markdown(SHARED_UI_WARNING)
+          
         with gr.Row():
                 
             with gr.Tab("Test") as inference_column: