Update app.py

app.py

@@ -42,8 +42,12 @@ pipe.unet.load_attn_procs(
     weight_name="pytorch_custom_diffusion_weights.bin"
 )
 
+print("Pipeline loaded")
+
 def generate(prompt, steps, guidance):
 
+    print("Generating...")
+    
     image = pipe(
         prompt,
         num_inference_steps=steps,
@@ -51,6 +55,8 @@ def generate(prompt, steps, guidance):
         eta=1
     ).images[0]
 
+    print("Done")
+    
     return image