Fix/studio (#947)

Author: tastelikefeet

swift/ui/app.py

@@ -31,6 +31,12 @@
     },
 }
 
+is_spaces = True if 'SPACE_ID' in os.environ else False
+if is_spaces:
+    is_shared_ui = True if 'modelscope/swift' in os.environ['SPACE_ID'] else False
+else:
+    is_shared_ui = False
+
 
 def run_ui():
     LLMTrain.set_lang(lang)
@@ -39,6 +45,16 @@ def run_ui():
         gr.HTML(f"<h1><center>{locale_dict['title'][lang]}</center></h1>")
         gr.HTML(f"<h3><center>{locale_dict['sub_title'][lang]}</center></h3>")
         gr.HTML(f"<h3><center>{locale_dict['star_beggar'][lang]}</center></h3>")
+        if is_shared_ui:
+            gr.HTML(f'''
+                <div class="gr-prose" style="max-width: 80%">
+                <p>If the waiting queue is too long, you can either run locally or duplicate the Space and run it on your own profile using a (paid) private A10G-large GPU for training. A A10G-large costs US$3.15/h. &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> # noqa
+                <img style="position: absolute; top: 0;right: 0; height: 100%;margin-top: 0px !important" src="file=duplicate.png"> # noqa
+                </div>
+            ''')
+        elif is_spaces and os.environ.get('MODELSCOPE_ENVIRONMENT') == 'studio':
+            gr.HTML('<p><center>You have duplicated the space, remember remove the `MODELSCOPE_ENVIRONMENT` '
+                    'variable to start an unlimited version</center></p>')
         with gr.Tabs():
             LLMTrain.build_ui(LLMTrain)
             LLMInfer.build_ui(LLMInfer)