Update app.py

app.py

@@ -19,6 +19,17 @@ def parse_args() -> argparse.Namespace:
                         action='store_false')
     return parser.parse_args()
 
+is_shared_ui = True if "AIFILMS/StyleGANEX" in os.environ['SPACE_ID'] else False
+
+if(is_shared_ui):
+    with gr.Box():
+        top_description = gr.HTML(f'''
+        <div class="gr-prose" style="max-width: 80%">
+            <h2 style="margin-top: 0">Attention - This Space doesn't work in this shared UI</h2>
+            <p>For it to work, you can access the <a href="https://huggingface.co/spaces/PKUWilliamYang/StyleGANEX">original</a> or duplicate this Space and run it on your own profile using a GPU.&nbsp;&nbsp;<a class="duplicate-button" style="display:inline-block" target="_blank" href="https://huggingface.co/spaces/{os.environ['SPACE_ID']}?duplicate=true"><img 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></p>
+        </div>
+        ''')
+
 DESCRIPTION = '''
 <div align=center>
 <h1 style="font-weight: 900; margin-bottom: 7px;">