Upload 5 files

feifeilib/feifeiflorence.py

@@ -7,6 +7,7 @@ from transformers import (
 )
 import torch
 import subprocess
+from feifeilib.feifeitexttoimg import feifeitexttoimg
 
 subprocess.run(
     "pip install flash-attn --no-build-isolation",
@@ -54,4 +55,6 @@ def feifeiflorence(
         generated_text,
         task=task_prompt,
         image_size=(image.width, image.height))
-    return parsed_answer["<MORE_DETAILED_CAPTION>"]
+    out_text=parsed_answer["<MORE_DETAILED_CAPTION>"]
+    out_img=feifeitexttoimg(out_text)
+    return out_text,out_img

feifeilib/feifeitexttoimg.py

@@ -20,27 +20,25 @@ MAX_SEED = np.iinfo(np.int32).max
 # 去除每行末尾的换行符
 artists = [artist.strip() for artist in artists]
 
-
 @spaces.GPU()
 def feifeitexttoimg(
         prompt,
-        quality_select,
-        sharpened_select,
-        styles_Radio,
-        FooocusExpansion_select,
-        seed=42,
+        quality_select=False,
+        sharpened_select=False,
+        styles_Radio=["(None)"],
+        FooocusExpansion_select=False,
+        seed=random.randint(0, MAX_SEED),
         randomize_seed=False,
-        width=1024,
-        height=1024,
+        width=896,
+        height=1152,
         num_inference_steps=4,
         guidance_scale=3.5,
         num_strength=0.35,
         progress=gr.Progress(track_tqdm=True),
 ):
-    guidance_scale = 3.5
+
     if randomize_seed:
         seed = random.randint(0, MAX_SEED)
-
     generator = torch.Generator().manual_seed(seed)
     if not prompt:
         prompt = "the photo is a 18 yo jpop girl is looking absolutely adorable and gorgeous, with a playful and mischievous grin, her eyes twinkling with joy. art by __artist__ and __artist__"

feifeiui/feifeiui.py

@@ -1,143 +1,150 @@
-import gradio as gr
-import numpy as np
-import config
-
-from feifeilib.feifeichat import feifeichat
-from feifeilib.feifeitexttoimg import feifeitexttoimg
-from feifeilib.feifeiflorence import feifeiflorence
-
-MAX_SEED = np.iinfo(np.int32).max
-MAX_IMAGE_SIZE = 4096
-
-css = """
-#col-container {
-    width: auto;
-    height: 750px;
-}
-"""
-
-
-def create_ui():
-    with gr.Blocks(css=css) as FeiFei:
-        with gr.Row():
-            with gr.Column(scale=1):
-                with gr.Tab("Generator"):
-                    prompt = gr.Text(
-                        label="Prompt",
-                        show_label=False,
-                        placeholder="Enter your prompt",
-                        max_lines=12,
-                        container=False,
-                    )
-                    run_button = gr.Button("Run")
-                    result = gr.Image(label="Result",
-                                      show_label=False,
-                                      interactive=False)
-
-                    with gr.Accordion("Advanced Settings", open=False):
-                        seed = gr.Slider(
-                            label="Seed",
-                            minimum=0,
-                            maximum=MAX_SEED,
-                            step=1,
-                            value=0,
-                        )
-
-                        randomize_seed = gr.Checkbox(label="Randomize seed",
-                                                     value=True)
-
-                        with gr.Row():
-                            width = gr.Slider(
-                                label="Width",
-                                minimum=256,
-                                maximum=MAX_IMAGE_SIZE,
-                                step=64,
-                                value=896,
-                            )
-
-                            height = gr.Slider(
-                                label="Height",
-                                minimum=256,
-                                maximum=MAX_IMAGE_SIZE,
-                                step=64,
-                                value=1152,
-                            )
-
-                        with gr.Row():
-                            num_inference_steps = gr.Slider(
-                                label="Number of inference steps",
-                                minimum=1,
-                                maximum=50,
-                                step=1,
-                                value=4,
-                            )
-                            guidancescale = gr.Slider(
-                                label="Guidance scale",
-                                minimum=0,
-                                maximum=10,
-                                step=0.1,
-                                value=3.5,
-                            )
-                            num_strength = gr.Slider(
-                                label="strength",
-                                minimum=0,
-                                maximum=2,
-                                step=0.01,
-                                value=0.35,
-                            )
-
-                with gr.Tab("Styles"):
-                    quality_select = gr.Checkbox(label="high quality")
-                    sharpened_select = gr.Checkbox(label="Sharpened")
-                    FooocusExpansion_select = gr.Checkbox(
-                        label="FooocusExpansion")
-                    styles_name = [
-                        style["name"] for style in config.style_list
-                    ]
-                    styles_Radio = gr.Dropdown(styles_name,
-                                               label="Styles",
-                                               multiselect=True)
-                with gr.Tab("Florence-2"):
-                    with gr.Row():
-                        with gr.Column():
-                            output_text = gr.Textbox(label="Output Text",
-                                                     max_lines=12,
-                                                     container=False)
-                            florence_btn = gr.Button(value="Florence")
-                        with gr.Column():
-                            input_img = gr.Image(label="Input Picture")
-            with gr.Column(scale=3, elem_id="col-container"):
-                gr.ChatInterface(
-                    feifeichat,
-                    type="messages",
-                    multimodal=True,
-                    additional_inputs=[
-                        gr.Checkbox(label="Feifei"),
-                    ],
-                )
-
-        run_button.click(
-            fn=feifeitexttoimg,  # Function to run for this button
-            inputs=[
-                prompt,
-                quality_select,
-                sharpened_select,
-                styles_Radio,
-                FooocusExpansion_select,
-                seed,
-                randomize_seed,
-                width,
-                height,
-                num_inference_steps,
-                guidancescale,
-                num_strength,
-            ],
-            outputs=[result, seed],
-        )
-
-        florence_btn.click(
-            fn=feifeiflorence,  # Function to run when the button is clicked
-            inputs=[input_img],  # Input components for the function
-            outputs=[output_text],  # Output component for the function
-        )
-    return FeiFei
+import gradio as gr
+import numpy as np
+import config
+
+from feifeilib.feifeichat import feifeichat
+from feifeilib.feifeitexttoimg import feifeitexttoimg
+from feifeilib.feifeiflorence import feifeiflorence
+
+MAX_SEED = np.iinfo(np.int32).max
+MAX_IMAGE_SIZE = 4096
+
+css = """
+#col-container {
+    width: auto;
+    height: 750px;
+}
+"""
+
+
+def create_ui():
+    with gr.Blocks(css=css) as FeiFei:
+        with gr.Row():
+            with gr.Column(scale=1):
+                with gr.Tab("Generator"):
+                    prompt = gr.Text(
+                        label="Prompt",
+                        show_label=False,
+                        placeholder="Enter your prompt",
+                        max_lines=12,
+                        container=False,
+                    )
+                    run_button = gr.Button("Run")
+                    result = gr.Image(label="Result",
+                                      show_label=False,
+                                      interactive=False)
+
+                    with gr.Accordion("Advanced Settings", open=False):
+                        seed = gr.Slider(
+                            label="Seed",
+                            minimum=0,
+                            maximum=MAX_SEED,
+                            step=1,
+                            value=0,
+                        )
+
+                        randomize_seed = gr.Checkbox(label="Randomize seed",
+                                                     value=True)
+
+                        with gr.Row():
+                            width = gr.Slider(
+                                label="Width",
+                                minimum=256,
+                                maximum=MAX_IMAGE_SIZE,
+                                step=64,
+                                value=896,
+                            )
+
+                            height = gr.Slider(
+                                label="Height",
+                                minimum=256,
+                                maximum=MAX_IMAGE_SIZE,
+                                step=64,
+                                value=1152,
+                            )
+
+                        with gr.Row():
+                            num_inference_steps = gr.Slider(
+                                label="Number of inference steps",
+                                minimum=1,
+                                maximum=50,
+                                step=1,
+                                value=4,
+                            )
+                            guidancescale = gr.Slider(
+                                label="Guidance scale",
+                                minimum=0,
+                                maximum=10,
+                                step=0.1,
+                                value=3.5,
+                            )
+                            num_strength = gr.Slider(
+                                label="strength",
+                                minimum=0,
+                                maximum=2,
+                                step=0.01,
+                                value=0.35,
+                            )
+
+                with gr.Tab("Styles"):
+                    quality_select = gr.Checkbox(label="high quality")
+                    sharpened_select = gr.Checkbox(label="Sharpened")
+                    FooocusExpansion_select = gr.Checkbox(
+                        label="FooocusExpansion")
+                    styles_name = [
+                        style["name"] for style in config.style_list
+                    ]
+                    styles_Radio = gr.Dropdown(styles_name,
+                                               label="Styles",
+                                               multiselect=True)
+                with gr.Tab("Florence-2"):
+                    with gr.Row():
+                        with gr.Column():
+                            output_text = gr.Textbox(label="Output Text",
+                                                     show_label=False,
+                                                     container=False)
+                            florence_btn = gr.Button(value="Florence")
+
+                        with gr.Column(scale=1):
+                            input_img = gr.Image(label="Input Picture",
+                                                 show_label=False)
+                        with gr.Column(scale=1):
+                            output_img = gr.Image(label="Input Picture",
+                                                  interactive=False,
+                                                  show_label=False)
+            with gr.Column(scale=3, elem_id="col-container"):
+                gr.ChatInterface(
+                    feifeichat,
+                    type="messages",
+                    multimodal=True,
+                    additional_inputs=[
+                        gr.Checkbox(label="Feifei"),
+                    ],
+                )
+
+        run_button.click(
+            fn=feifeitexttoimg,  # Function to run for this button
+            inputs=[
+                prompt,
+                quality_select,
+                sharpened_select,
+                styles_Radio,
+                FooocusExpansion_select,
+                seed,
+                randomize_seed,
+                width,
+                height,
+                num_inference_steps,
+                guidancescale,
+                num_strength,
+            ],
+            outputs=[result, seed],
+        )
+
+        florence_btn.click(
+            fn=feifeiflorence,  # Function to run when the button is clicked
+            inputs=[input_img],  # Input components for the function
+            outputs=[output_text,
+                     output_img],  # Output component for the function
+        )
+    return FeiFei