wew

.gitignore

@@ -1,6 +1,7 @@
-_pycache_
-_pycache_/*
+__pycache__
+__pycache__/*
 ./_pycache_/*
 __pycache__/controlnet_flux.cpython-310.pyc
 __pycache__/pipeline_flux_controlnet_inpaint.cpython-310.pyc
 __pycache__/transformer_flux.cpython-310.pyc
+__pycache__/pipeline_flux_cnet.cpython-310.pyc

app.py

@@ -10,7 +10,7 @@ from PIL import Image, ImageDraw
 import numpy as np
 import subprocess
 
-subprocess.run("rm -rf /data-nvme/zerogpu-offload/*", env={}, shell=True)
+# subprocess.run("rm -rf /data-nvme/zerogpu-offload/*", env={}, shell=True)
 
 
 HF_TOKEN = os.getenv("HF_TOKEN")
@@ -92,7 +92,7 @@ def create_diptych_image(image):
     return diptych
 
 @spaces.GPU()
-def inpaint_image(image, prompt, editor_value):
+def inpaint_image(image, prompt, subject, editor_value):
     # Load image and mask
     size = (1536, 768)
     image = load_image(image).convert("RGB").resize((768, 768))
@@ -125,10 +125,13 @@ def inpaint_image(image, prompt, editor_value):
     # Convert to bfloat16 to match model dtype
     full_attn_scale_mask = full_attn_scale_mask.to(device=pipe.transformer.device, dtype=torch.bfloat16)
 
+    subject_name=subject
+    target_text_prompt=prompt
+    prompt_final=f'A two side-by-side image of same {subject_name}. LEFT: a photo of the {subject_name}; RIGHT: a photo of the {subject_name} {target_text_prompt}.'
 
     # Inpaint
     result = pipe(
-        prompt=prompt,
+        prompt=prompt_final,
         height=size[1],
         width=size[0],
         control_image=diptych_image,
@@ -149,6 +152,7 @@ iface = gr.Interface(
     inputs=[
         gr.Image(type="filepath", label="Upload Image"),
         gr.Textbox(lines=2, placeholder="Enter your prompt here (e.g., 'wearing a christmas hat, in a busy street')", label="Prompt"),
+        gr.Textbox(lines=1, placeholder="Enter your subject", label="Subject"),
         gr.ImageEditor(type="pil", label="Image with Mask", sources="upload", interactive=True)
     ],
     outputs=[