Update app.py

app.py

@@ -157,6 +157,7 @@ prng_seed = jax.random.split(rng, jax.device_count())
 memory = psutil.virtual_memory()
 
 def infer(prompt, negative_prompt, image, model_type="Standard"):
+    time_start = time.time()
     prompts = num_samples * [prompt]
     if model_type=="Standard":
         prompt_ids = std_pipeline.prepare_text_inputs(prompts)