Update app.py

app.py

@@ -6,9 +6,6 @@ import torch
 from diffusers import DiffusionPipeline, FlowMatchEulerDiscreteScheduler
 from PIL import Image
 import io
-import os
-import subprocess
-subprocess.run('pip install flash-attn --no-build-isolation', env={'FLASH_ATTENTION_SKIP_CUDA_BUILD': "TRUE"}, shell=True)
 
 dtype = torch.bfloat16
 device = "cuda" if torch.cuda.is_available() else "cpu"
@@ -191,24 +188,6 @@ with gr.Blocks(css=css, theme=gr.themes.Soft(primary_hue="blue", secondary_hue="
             cache_examples="lazy"
         )
 
-        share_buttons = gr.Row()
-        with share_buttons:
-            twitter_button = gr.Button("Share on Twitter")
-            facebook_button = gr.Button("Share on Facebook")
-
-        twitter_button.click(
-            fn=lambda img: f"https://twitter.com/intent/tweet?url={img}",
-            inputs=[result],
-            outputs=None,
-            _js="(img) => window.open(img, '_blank')"
-        )
-
-        facebook_button.click(
-            fn=lambda img: f"https://www.facebook.com/sharer/sharer.php?u={img}",
-            inputs=[result],
-            outputs=None,
-            _js="(img) => window.open(img, '_blank')"
-        )
 
     gr.on(
         triggers=[run_button.click, prompt.submit],
@@ -220,16 +199,7 @@ with gr.Blocks(css=css, theme=gr.themes.Soft(primary_hue="blue", secondary_hue="
     demo.load(
         fn=lambda: None,
         inputs=None,
-        outputs=None,
-        _js="""
-        function() {
-            document.addEventListener('keydown', function(event) {
-                if (event.key === 'Enter') {
-                    document.querySelector('button').click();
-                }
-            });
-        }
-        """
+        outputs=None
     )