Update app.py

app.py

@@ -41,15 +41,19 @@ def caption2hash(caption: str) -> str:
 
 # ---- Inference function ----
 def infer(caption: str, condition_image: Image.Image, steps: int = 20, seed: int = 0, invert: bool = False):
+    print("Loading condition image")
     img = condition_image.convert("RGB")
     if invert:
         img = ImageOps.invert(img)
+        print("Condition image inverted")
     cache_file = Path(f"inferences/{pil2hash(img)}_{caption2hash(caption)}.png")
     if cache_file.exists():
         return Image.open(cache_file)
 
     generator = torch.manual_seed(seed)
+    print("Starting generation...")
     output = pipe(prompt=caption, image=img, num_inference_steps=steps, generator=generator).images[0]
+    print("Caching result...")
     output.save(cache_file)
     return output