Update app.py

app.py

@@ -211,6 +211,9 @@ def randomize_seed_fn(seed: int, randomize_seed: bool) -> int:
     return seed
 
 def uploadNote():
+          # write note txt
+    filename= f'tst_B_{seed}.txt'
+    timestamp = datetime.datetime.now().strftime("%Y%m%d_%H%M%S")
     with open(filename, "w") as f:
         f.write(f"Realvis 5.0 (Tester B): {seed} png\n")
         f.write(f"Date/time: {timestamp} \n")
@@ -265,10 +268,7 @@ def generate_30(
         options["use_resolution_binning"] = True
     images = []
     pipe.scheduler.set_timesteps(num_inference_steps,device)
-      # write note txt
-    filename= f'tst_{seed}.txt'
-    timestamp = datetime.datetime.now().strftime("%Y%m%d_%H%M%S")
-
+    uploadNote()
     for i in range(0, num_images, BATCH_SIZE):
         batch_options = options.copy()
         batch_options["prompt"] = options["prompt"][i:i+BATCH_SIZE]
@@ -323,9 +323,6 @@ def generate_60(
         options["use_resolution_binning"] = True
     images = []
     pipe.scheduler.set_timesteps(num_inference_steps,device)
-      # write note txt
-    filename= f'tst_{seed}.txt'
-    timestamp = datetime.datetime.now().strftime("%Y%m%d_%H%M%S")
     uploadNote()
     for i in range(0, num_images, BATCH_SIZE):
         batch_options = options.copy()
@@ -381,9 +378,6 @@ def generate_90(
         options["use_resolution_binning"] = True
     images = []
     pipe.scheduler.set_timesteps(num_inference_steps,device)
-      # write note txt
-    filename= f'tst_{seed}.txt'
-    timestamp = datetime.datetime.now().strftime("%Y%m%d_%H%M%S")
     uploadNote()
     for i in range(0, num_images, BATCH_SIZE):
         batch_options = options.copy()