Update app.py

app.py

@@ -350,7 +350,8 @@ def generate_30(
     samples=1,
     progress=gr.Progress(track_tqdm=True)  # Add progress as a keyword argument
 ):
-    captioner_2=captioner_2
+    global captioner_2
+    captioner2=captioner_2
     seed = random.randint(0, MAX_SEED)
     generator = torch.Generator(device='cuda').manual_seed(seed)
     if latent_file is not None:  # Check if a latent file is provided
@@ -359,14 +360,14 @@ def generate_30(
         caption=[]
         caption_2=[]
         #caption.append(captioner(sd_image_a))
-        caption.append(captioner_2(sd_image_a))
+        caption.append(captioner2(sd_image_a))
         #caption.append(captioner_3(sd_image_a))
         caption_2.append(captioning(sd_image_a))
         if latent_file_2 is not None:  # Check if a latent file is provided
             sd_image_b = Image.open(latent_file_2.name).convert('RGB')
             sd_image_b.resize((height,width), Image.LANCZOS)
             #caption.append(captioner(sd_image_b))
-            caption.append(captioner_2(sd_image_b))
+            caption.append(captioner2(sd_image_b))
             #caption.append(captioner_3(sd_image_b))
             caption_2.append(captioning(sd_image_b))
         else:
@@ -375,7 +376,7 @@ def generate_30(
             sd_image_c = Image.open(latent_file_3.name).convert('RGB')
             sd_image_c.resize((height,width), Image.LANCZOS)
             #caption.append(captioner(sd_image_c))
-            caption.append(captioner_2(sd_image_c))
+            caption.append(captioner2(sd_image_c))
             #caption.append(captioner_3(sd_image_c))
             caption_2.append(captioning(sd_image_c))
         else:
@@ -384,7 +385,7 @@ def generate_30(
             sd_image_d = Image.open(latent_file_4.name).convert('RGB')
             sd_image_d.resize((height,width), Image.LANCZOS)
             #caption.append(captioner(sd_image_d))
-            caption.append(captioner_2(sd_image_d))
+            caption.append(captioner2(sd_image_d))
             #caption.append(captioner_3(sd_image_d))
             caption_2.append(captioning(sd_image_d))
         else:
@@ -393,7 +394,7 @@ def generate_30(
             sd_image_e = Image.open(latent_file_5.name).convert('RGB')
             sd_image_e.resize((height,width), Image.LANCZOS)
             #caption.append(captioner(sd_image_e))
-            caption.append(captioner_2(sd_image_e))
+            caption.append(captioner2(sd_image_e))
             #caption.append(captioner_3(sd_image_e))
             caption_2.append(captioning(sd_image_e))
         else:
@@ -404,7 +405,7 @@ def generate_30(
         print(caption)
         print(caption_2)
         print("-- generating further caption --")
-        del captioner_2
+        del captioner2
         del model5
         del processor5
         gc.collect()