app

app.py

@@ -21,7 +21,7 @@ from ip_adapter import CSGO
 from transformers import BlipProcessor, BlipForConditionalGeneration
 
 device = "cuda" if torch.cuda.is_available() else "cpu"
-
+dtype = torch.float16 if str(device).__contains__("cuda") else torch.float32
 import os
 os.system("git lfs install")
 os.system("git clone https://huggingface.co/h94/IP-Adapter")
@@ -101,20 +101,28 @@ def get_example():
             "there is a small house with a sheep statue on top of it",
             0.6,
             1.0,
+            7.0,
+            42
         ],
         [
          None,
          './assets/img_1.png',
             "Text-Driven Style Synthesis",
          "a cat",
-         0.01,1.0
+         0.01,
+            1.0,
+            7.0,
+            42
          ],
         [
             None,
             './assets/img_2.png',
             "Text-Driven Style Synthesis",
             "a building",
-            0.01, 1.0
+            0.01,
+            1.0,
+            7.0,
+            42,
         ],
         [
             "./assets/img_0.png",
@@ -122,23 +130,25 @@ def get_example():
             "Text Edit-Driven Style Synthesis",
             "there is a small house",
             0.4,
-            1.0
+            1.0,
+            7.0,
+            42,
         ],
     ]
     return case
 
-@spaces.GPU(duration=150)
-def run_for_examples(content_image_pil,style_image_pil,target, prompt, scale_c, scale_s):
+
+def run_for_examples(content_image_pil,style_image_pil,target, prompt, scale_c, scale_s,guidance_scale,seed):
     return create_image(
         content_image_pil=content_image_pil,
         style_image_pil=style_image_pil,
         prompt=prompt,
         scale_c=scale_c,
         scale_s=scale_s,
-        guidance_scale=7.0,
+        guidance_scale=guidance_scale,
         num_samples=2,
         num_inference_steps=50,
-        seed=42,
+        seed=seed,
         target=target,
     )
 def randomize_seed_fn(seed: int, randomize_seed: bool) -> int:
@@ -156,7 +166,7 @@ def image_grid(imgs, rows, cols):
     for i, img in enumerate(imgs):
         grid.paste(img, box=(i % cols * w, i // cols * h))
     return grid
-@spaces.GPU(duration=150)
+@spaces.GPU(duration=200)
 def create_image(content_image_pil,
                  style_image_pil,
                  prompt,
@@ -350,10 +360,10 @@ with block:
 
     gr.Examples(
         examples=get_example(),
-        inputs=[content_image_pil,style_image_pil,target, prompt, scale_c, scale_s],
+        inputs=[content_image_pil,style_image_pil,target, prompt, scale_c, scale_s,guidance_scale,seed],
         fn=run_for_examples,
         outputs=[generated_image],
-        cache_examples=True,
+        cache_examples=False,
     )
 
     gr.Markdown(article)