Update app.py

app.py

@@ -11,6 +11,8 @@ import pathlib
 import shlex
 import subprocess
 
+is_shared_ui = True if "AIFILMS/ControlNet-Video" in os.environ['SPACE_ID'] else False
+
 if os.getenv('SYSTEM') == 'spaces':
     with open('patch') as f:
         subprocess.run(shlex.split('patch -p1'), stdin=f, cwd='ControlNet')
@@ -34,10 +36,13 @@ for name in names:
         continue
     subprocess.run(shlex.split(command), cwd='ControlNet/annotator/ckpts/')
 
-from model import (DEFAULT_BASE_MODEL_FILENAME, DEFAULT_BASE_MODEL_REPO,
-                   DEFAULT_BASE_MODEL_URL, Model)
 
-model = Model()
+
+if(not is_shared_ui):
+    from model import (DEFAULT_BASE_MODEL_FILENAME, DEFAULT_BASE_MODEL_REPO,
+                       DEFAULT_BASE_MODEL_URL, Model)
+    
+    model = Model()
 
 
 def controlnet(i, prompt, control_task, seed_in, ddim_steps, scale, low_threshold, high_threshold, value_threshold, distance_threshold, bg_threshold):
@@ -155,6 +160,8 @@ def create_video(frames, fps, type):
 
 
 def infer(prompt,video_in, control_task, seed_in, trim_value, ddim_steps, scale, low_threshold, high_threshold, value_threshold, distance_threshold, bg_threshold, gif_import):
+    if(is_shared_ui):
+        raise gr.Error("This Space doesn't work on this shared UI.")
     print(f"""
     ———————————————
     {prompt}
@@ -248,6 +255,14 @@ article = """
 """
 
 with gr.Blocks(css='style.css') as demo:
+    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/fffiloni/ControlNet-Video">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>
+            ''')
     with gr.Column(elem_id="col-container"):
         gr.HTML(title)
         gr.HTML("""
@@ -301,30 +316,30 @@ with gr.Blocks(css='style.css') as demo:
 
                     with gr.Tab("Custom Model"):
                         current_base_model = gr.Text(label='Current base model',
-                                             value=DEFAULT_BASE_MODEL_URL)
+                                             value="" if is_shared_ui else DEFAULT_BASE_MODEL_URL)
                         with gr.Row():
                             with gr.Column():
                                 base_model_repo = gr.Text(label='Base model repo',
                                                       max_lines=1,
-                                                      placeholder=DEFAULT_BASE_MODEL_REPO,
+                                                      placeholder="" if is_shared_ui else DEFAULT_BASE_MODEL_REPO,
                                                       interactive=True)
                                 base_model_filename = gr.Text(
                                      label='Base model file',
                                      max_lines=1,
-                                     placeholder=DEFAULT_BASE_MODEL_FILENAME,
+                                     placeholder="" if is_shared_ui else DEFAULT_BASE_MODEL_FILENAME,
                                      interactive=True)
                             change_base_model_button = gr.Button('Change base model')
                         
                         gr.HTML(
                             '''<p>You can use other base models by specifying the repository name and filename.<br />
                                   The base model must be compatible with Stable Diffusion v1.5.</p>''')
-                
-                        change_base_model_button.click(fn=model.set_base_model,
-                                                       inputs=[
-                                                           base_model_repo,
-                                                           base_model_filename,
-                                                       ],
-                                                       outputs=current_base_model, queue=False)
+                        if(not is_shared_ui):
+                            change_base_model_button.click(fn=model.set_base_model,
+                                                           inputs=[
+                                                               base_model_repo,
+                                                               base_model_filename,
+                                                           ],
+                                                           outputs=current_base_model, queue=False)
                 
                 submit_btn = gr.Button("Generate ControlNet video")