Update injection_main_HF.py

injection_main_HF.py

@@ -585,16 +585,25 @@ if __name__ == "__main__":
             image = transforms.ToPILImage()(image[0])
 
             examples.append([entry["target_prompt"], image, None, None, None])
-
-       
-        
-        
-        
-        
+ 
         with gr.Blocks() as app:
-
+        
             with gr.Column():
-                gr.Markdown("Artist Interactive Demo")
+                gr.Markdown("# Artist: Aesthetically Controllable Text-Driven Stylization without Training")
+                gr.Markdown("## Interactive Demo, HF space version")
+                gr.HTML("""
+                <div style="display:flex;column-gap:4px;">
+                    <a href='https://diffusionartist.github.io/'>
+                        <img src='https://img.shields.io/badge/Project-Page-green'>
+                    </a>
+                    <a href='https://github.com/songrise/Artist'>
+                        <img src='https://img.shields.io/badge/Code-github-blue'>
+                    </a>
+                    <a href='https://arxiv.org/abs/2407.15842'>
+                        <img src='https://img.shields.io/badge/Paper-Arxiv-red'>
+                    </a>
+                </div>
+                """)
                 with gr.Row():
                     with gr.Column():
                         text_input = gr.Textbox(