Update app_inpaint2.py

app_inpaint2.py

@@ -58,38 +58,35 @@ def load_default_pipeline():
         vae=vae,
         controlnet=model,
     ).to("cuda")
-    # 此函数原用于 Misc 选项卡，现在无需返回 gr.update
     print("Default pipeline loaded!")
 
 @spaces.GPU(duration=7)
 def fill_image(prompt, image, model_selection, paste_back):
     """
-    处理 ImageMask（gr.ImageMask）输入的 fill/repair 流程。
-    在这里对用户绘制的 mask 做默认 5% 膨胀。
+    Handles the fill/repair process for inputs from ImageMask (gr. ImageMask). Applies a default 5% expansion to user-drawn masks here.
     """
     print(f"Received image: {image}")
     if image is None:
         yield None, None
         return
 
-    # 如果用户选择了不同的模型 key，则加载对应预训练仓库
-    # 注意：此逻辑原 Outpaint 中有，Inpaint 中缺失，现在补充以支持模型切换
-    global pipe
-    if model_selection in MODELS and pipe.config.model_name != MODELS[model_selection]:
-        # 释放旧模型显存
-        del pipe
-        torch.cuda.empty_cache()
-        pipe = StableDiffusionXLFillPipeline.from_pretrained(
-            MODELS[model_selection],
-            torch_dtype=torch.float16,
-            vae=vae,
-            controlnet=model,
-            variant="fp16", # 保持 variant 设置
-        )
-        pipe.scheduler = TCDScheduler.from_config(pipe.scheduler.config)
-        pipe.to("cuda")
-        print(f"Loaded new SDXL model: {pipe.config.model_name}")
-
+    if model_selection in MODELS:
+        current_model = pipe.config.get("_name_or_path", "")
+        target_model = MODELS[model_selection]
+        if current_model != target_model:
+            # 释放旧模型显存
+            del pipe
+            torch.cuda.empty_cache()
+            pipe = StableDiffusionXLFillPipeline.from_pretrained(
+                target_model,
+                torch_dtype=torch.float16,
+                vae=vae,
+                controlnet=model,
+                variant="fp16",
+            )
+            pipe.scheduler = TCDScheduler.from_config(pipe.scheduler.config)
+            pipe.to("cuda")
+            print(f"Loaded new SDXL model: {target_model}")
 
     (
         prompt_embeds,
@@ -157,16 +154,10 @@ def clear_result():
 
 def use_output_as_input(output_image):
     """
-    接收 ImageSlider 的输出 (image_out, cnet_image)
-    返回 cnet_image 作为新的输入。
+    Receives the output of ImageSlider (image_out, cnet_image) and returns cnet_image as the new input.
     """
-    # ImageSlider 的 value 是一个 tuple (image1, image2)
-    # 这里的 image2 (即 cnet_image) 是包含修复结果的图像 (如果 paste_back 为 True)
-    # 或者只是 ControlNet 输入（如果 fill_image 逻辑有变）
-    # 假设我们想要将修复后的结果图作为新的输入图像（新的 source）
-    # 在 fill_image 中，最终 yield 的是 (final_output, cnet_image)
-    # 我们应该使用 final_output 作为新的背景图。
-    return gr.update(value=output_image[0]) # output_image[0] 是最终修复图像
+
+    return gr.update(value=output_image[0])
 
 css = """
 .nulgradio-container {
@@ -176,7 +167,7 @@ css = """
     overflow-y: scroll !important;
     padding: 10px 40px !important;
 }
-div#component-17 { /* 这是一个动态 ID，可能需要调整或移除 */
+div#component-17 {
     height: auto !important;
 }
 
@@ -185,7 +176,6 @@ div#component-17 { /* 这是一个动态 ID，可能需要调整或移除 */
         display: block !important;
         margin-bottom: 20px !important;
     }
-    /* 移除掉 component-16 的引用，因为它是动态的 */
 }
 
 """
@@ -202,9 +192,8 @@ title = """<h1 align="center">Diffusers Image Inpaint</h1>
 """
 
 with gr.Blocks(css=css, fill_height=True) as demo:
-    gr.Markdown(title) # 使用更简洁的 Markdown 标题
+    gr.Markdown(title)
     
-    # 只保留 Inpaint 选项卡的内容，移除 Tabs 结构，让 Inpaint 成为主界面
     with gr.Column():
         with gr.Row():
             with gr.Column():
@@ -238,7 +227,7 @@ with gr.Blocks(css=css, fill_height=True) as demo:
             fn=use_output_as_input, 
             inputs=[result], 
             outputs=[input_image],
-            queue=False # 这是一个快速操作
+            queue=False
         )
         
         # Generates image on button click
@@ -285,5 +274,4 @@ with gr.Blocks(css=css, fill_height=True) as demo:
             queue=False,
         )
 
-# 将 queue 和 launch 保持不变
 demo.queue(max_size=10).launch(show_error=True)