Update app.py

app.py

@@ -59,12 +59,12 @@ class calculateDuration:
 @spaces.GPU(duration=120)
 @torch.inference_mode()
 def generate_image(prompt, steps, seed, cfg_scale, width, height, progress):
-    
-    with calculateDuration("Generating image"):
-        # Generate image
+    with calculateDuration(f"Make a new generator:${seed}"):
         pipe.to(device)
         generator = torch.Generator(device=device).manual_seed(seed)
-        print(prompt, steps, seed, cfg_scale, width, height)
+        
+    with calculateDuration("Generating image"):
+        # Generate image
         generated_image = pipe(
             prompt=prompt,
             num_inference_steps=steps,
@@ -74,7 +74,7 @@ def generate_image(prompt, steps, seed, cfg_scale, width, height, progress):
             generator=generator,
         ).images[0]
     
-    progress(99, "Generate success!")
+    progress(99, "Generate image success!")
     return generated_image
 
 
@@ -135,9 +135,11 @@ def run_lora(prompt, lora_strings_json,  cfg_scale, steps, randomize_seed, seed,
     # Generate image
     error_message = ""
     try:
+        print("Start applying for zeroGPU resources")
         final_image = generate_image(prompt, steps, seed, cfg_scale, width, height, progress)
     except Exception as e:
         error_message =  str(e)
+        print("Run error", e)
         final_image = None
         
     if final_image: