update app

app.py

@@ -6,6 +6,7 @@ import random
 import torch
 import subprocess
 from diffusers import StableDiffusionPipeline
+import time
 
 model_id = "dicoo_model"
 
@@ -14,7 +15,9 @@ pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float
 def predict(prompt, steps=30, seed=42, guidance_scale=7.5):
     print("prompt: ", prompt)
     print("steps: ", steps)
+    start_time = time.time()
     image = pipe(prompt, num_inference_steps=steps, guidance_scale=7.5).images[0]
+    print("cost: ", time.time() - start_time)
     return image
 
 md = """