Update app.py

app.py

@@ -13,7 +13,15 @@ from subprocess import getoutput
 is_shared_ui = True if "fffiloni/MimicMotion" in os.environ['SPACE_ID'] else False
 available_property = False if is_shared_ui else True
 
-
+is_gpu_associated = torch.cuda.is_available()
+if is_gpu_associated:
+    gpu_info = getoutput('nvidia-smi')
+    if("A10G" in gpu_info):
+        which_gpu = "A10G"
+    elif("T4" in gpu_info):
+        which_gpu = "T4"
+    else:
+        which_gpu = "CPU"
 
 def stream_output(pipe):
     for line in iter(pipe.readline, ''):
@@ -240,21 +248,54 @@ div#warning-ready > .gr-prose > h2, div#warning-ready > .gr-prose > p {
     color: #030303 !important;
 }
 """
-with gr.Blocks() as demo:
-    with gr.Row():
-        ref_image_in = gr.Image()
-        ref_video_in = gr.Video()
-    num_inference_steps = gr.Slider(minimum=12, maximum=50, value=25, step=1)
-    guidance_scale = gr.Slider(minimum=0.1, maximum=10, value=2, step=0.1)
-    output_frames_per_second = gr.Slider(minimum=1, maximum=60, value=16, step=1)
-    seed = gr.Number(value=42)
-    checkpoint_version = gr.Dropdown(choices=["MimicMotion_1.pth", "MimicMotion_1-1.pth"], value="MimicMotion_1.pth")
-    submit_btn = gr.Button("실행")
-    
+with gr.Blocks(css=css) as demo:
+    with gr.Column():
+        gr.Markdown("# Mimic")
+        gr.Markdown("High-quality")
+        gr.HTML("""
+1
+        """)
+        with gr.Row():
+            with gr.Column():
+                if is_shared_ui:
+                    top_description = gr.HTML(f'''
+2
+                    ''', elem_id="warning-duplicate")
+                else:
+                    if(is_gpu_associated):
+                        top_description = gr.HTML(f'''
+2
+                        ''', elem_id="warning-ready")
+                    else:
+                        top_description = gr.HTML(f'''
+2
+                        ''', elem_id="warning-setgpu")
+                with gr.Row():
+                    ref_image_in = gr.Image(label="Person Image Reference", type="filepath")
+                    ref_video_in = gr.Video(label="Person Video Reference")
+                with gr.Accordion("Advanced Settings", open=False):
+                    num_inference_steps = gr.Slider(label="num inference steps", minimum=12, maximum=50, value=25, step=1, interactive=available_property)
+                    guidance_scale = gr.Slider(label="guidance scale", minimum=0.1, maximum=10, value=2, step=0.1, interactive=available_property)
+                    with gr.Row():
+                        output_frames_per_second = gr.Slider(label="fps", minimum=1, maximum=60, value=16, step=1, interactive=available_property)
+                        seed = gr.Number(label="Seed", value=42, interactive=available_property)
+                    checkpoint_version = gr.Dropdown(label="Checkpoint Version", choices=["MimicMotion_1.pth", "MimicMotion_1-1.pth"], value="MimicMotion_1.pth", interactive=available_property, filterable=False)
+                submit_btn = gr.Button("Submit", interactive=available_property)
+                gr.Examples(
+                    examples = [
+                        ["./examples/demo1.jpg", "./examples/preview_1.mp4"]
+                    ],
+                    fn = load_examples,
+                    inputs = [ref_image_in, ref_video_in],
+                    outputs = [output_video],
+                    run_on_click = True,
+                    cache_examples = False
+                )
+            output_video.render()
     submit_btn.click(
-        fn=infer,
-        inputs=[ref_image_in, ref_video_in, num_inference_steps, guidance_scale, output_frames_per_second, seed, checkpoint_version],
-        outputs=[gr.Video()]
+        fn = infer,
+        inputs = [ref_image_in, ref_video_in, num_inference_steps, guidance_scale, output_frames_per_second, seed, checkpoint_version],
+        outputs = [output_video]
     )
 
-demo.launch()
+demo.launch(show_api=False, show_error=False)