commit

app.py

@@ -20,12 +20,10 @@ from functools import partial
 import spaces
 import torch
 import gradio as gr
-from numpy import asarray as array
 from lxml.html import fromstring
-from diffusers.utils import export_to_video, load_image
 from huggingface_hub import hf_hub_download
 from safetensors.torch import load_file, save_file
-from diffusers import FluxPipeline, CogVideoXImageToVideoPipeline
+from diffusers import FluxPipeline
 from PIL import Image, ImageDraw, ImageFont
 
 # logging
@@ -48,10 +46,6 @@ else:
 
 base = "black-forest-labs/FLUX.1-schnell"
 
-# variable data
-
-
-
 # precision data
 
 seq=512
@@ -187,7 +181,9 @@ def handle_generate(artist,song,genre,lyrics):
     img.save(name)
     return name
 
-def ui():
+# entry
+
+if __name__ == "__main__":
     with gr.Blocks(theme=gr.themes.Citrus(),css=css,js=js) as demo:
         gr.Markdown(f"""
             # Song Cover Image Generator
@@ -219,22 +215,14 @@ def ui():
             )
         with gr.Row():
             run = gr.Button("Generate",elem_classes="btn")
+            run.click(
+                fn=handle_generate,
+                inputs=[artist,song,genre,lyrics],
+                outputs=[cover]
+            )
         with gr.Row():
             cover = gr.Image(interactive=False,container=False,elem_classes="image-container", label="Result", show_label=True, type='filepath', show_share_button=False)
 
-        gr.on(
-            triggers=[
-                run.click
-            ],
-            fn=handle_generate,
-            inputs=[artist,song,genre,lyrics],
-            outputs=[cover]
-        )
-        demo.queue().launch()
-
-# entry
-
-if __name__ == "__main__":
-    ui()
+    demo.queue().launch()
 
 # end