add examples (#3)

- Upload 8 files (9b6844a2a61f8eedf7509d8fa2077501fb7352ea)
- Update app.py (abd3b00d363f6c8f7cda2c601e658f3e49466db7)


Co-authored-by: Linoy Tsaban <[email protected]>

.gitattributes

@@ -33,3 +33,8 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zip filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
+examples[[:space:]]2/celeb_meme.jpg filter=lfs diff=lfs merge=lfs -text
+examples[[:space:]]2/cookie.png filter=lfs diff=lfs merge=lfs -text
+examples[[:space:]]2/leather.jpg filter=lfs diff=lfs merge=lfs -text
+examples[[:space:]]2/no_cookie.png filter=lfs diff=lfs merge=lfs -text
+examples[[:space:]]2/poster_orig.jpg filter=lfs diff=lfs merge=lfs -text

app.py

@@ -26,6 +26,17 @@ from modules.model_edit import Step1XParams, Step1XEdit
 
 print("TORCH_CUDA", torch.cuda.is_available())
 
+examples = [
+["examples 2/meme.jpg", "turn into an illustration in studio ghibli style",("examples 2/meme.jpg","examples 2/ghibli_meme.jpg"),],
+["examples 2/celeb_meme.jpg", "replace the gray blazer with a leather jacket",("examples 2/celeb_meme.jpg","examples 2/leather.jpg")],
+["examples 2/cookie.png", "remove the cookie",("examples 2/cookie.png","examples 2/no_cookie.png")],
+["examples 2/poster_orig.jpg", "replace 'lambs' with 'llamas'",("examples 2/poster_orig.jpg","examples 2/poster.jpg")],
+]
+
+def generate_examples(init_image, output_image, prompt):
+    return init_image, output_image, prompt
+
+
 def load_state_dict(model, ckpt_path, device="cuda", strict=False, assign=True):
     if Path(ckpt_path).suffix == ".safetensors":
         state_dict = load_file(ckpt_path, device)
@@ -444,7 +455,7 @@ with gr.Blocks() as demo:
     with gr.Row():
         with gr.Column():
             prompt = gr.Textbox(
-                label="编辑指令",
+                label="编辑指令 prompt",
                 value='Remove the person from the image.',
             )
             init_image = gr.Image(label="Input Image", type='pil')
@@ -470,4 +481,12 @@ with gr.Blocks() as demo:
         outputs=[output_image, output_random_seed],
     )
 
+    gr.Examples(
+        examples,
+        inputs=[init_image, prompt, output_image],
+        outputs=[init_image, prompt, output_image],
+        fn=generate_examples,
+        cache_examples=False
+        )
+
 demo.launch()