Update app.py

app.py

@@ -3,8 +3,7 @@ import gradio as gr
 import numpy as np
 import spaces
 import torch
-import os
-from huggingface_hub import hf_hub_download
+import re
 from diffusers import DiffusionPipeline
 from compel import Compel, ReturnedEmbeddingsType
 from PIL import Image, PngImagePlugin
@@ -34,7 +33,18 @@ def add_comma_after_pattern_ti(text):
     pattern = re.compile(r'\b\w+_\d+\b')
     modified_text = pattern.sub(lambda x: x.group() + ',', text)
     return modified_text
-    
+
+def get_embed_new(prompt, pipe, compel, only_convert_string=False):
+    """处理提示词的函数"""
+    if only_convert_string:
+        # 简单处理，添加逗号分隔
+        return add_comma_after_pattern_ti(prompt)
+    else:
+        # 如果需要更复杂的处理，可以在这里添加
+        return prompt
+
+DESCRIPTION = "梦羽的模型生成器 - 快速生成 MiaomiaoHarem vPred Dogma 1.1 模型的图片"
+
 if not torch.cuda.is_available():
     DESCRIPTION += "\n<p>你现在运行在CPU上 但是此项目只支持GPU.</p>"
 
@@ -62,7 +72,6 @@ def infer(
     num_inference_steps: int = 30,
     randomize_seed: bool = True,
     use_resolution_binning: bool = True,
-    progress=gr.Progress(track_tqdm=True),
 ):
     seed = int(randomize_seed_fn(seed, randomize_seed))
     generator = torch.Generator().manual_seed(seed)
@@ -107,7 +116,7 @@ def infer(
         "height": height,
         "guidance_scale": guidance_scale,
         "num_inference_steps": num_inference_steps,
-        "model": "miaomiaoHarem_vPredDogma11",
+        "model": "qwen-image",
         "use_resolution_binning": use_resolution_binning,
         "PreUrl": "https://huggingface.co/spaces/Menyu/miaomiaoHaremDogma11"
     }
@@ -132,7 +141,7 @@ h1{text-align:center}
 
 with gr.Blocks(css=css) as demo:
     gr.Markdown("""# 梦羽的模型生成器
-        ### 快速生成 MiaomiaoHarem vPred Dogma 1.1 模型的图片""")
+        ### 快速生成 qwen-image 模型的图片""")
     with gr.Group():
         with gr.Row():
             prompt = gr.Text(
@@ -193,6 +202,7 @@ with gr.Blocks(css=css) as demo:
                 step=1,
                 value=28,
             )
+        use_resolution_binning = gr.Checkbox(label="使用分辨率分箱", value=True)
 
     gr.Examples(
         examples=examples,
@@ -220,6 +230,7 @@ with gr.Blocks(css=css) as demo:
             guidance_scale,
             num_inference_steps,
             randomize_seed,
+            use_resolution_binning,
         ],
         outputs=[result, seed],
     )