Update app.py

app.py

@@ -427,6 +427,11 @@ def generate_30(
         expanded_1 = expanded[0]
         expanded_2 = expanded[1]
         new_prompt = prompt+' '+expanded_1
+        print("-- ------------ --")
+        print("-- FINAL PROMPT --")
+        print(new_prompt)
+        print("-- FINAL PROMPT --")
+        print("-- ------------ --")
         global model
         global txt_tokenizer
         del model
@@ -569,18 +574,15 @@ def generate_60(
         del processor5
         gc.collect()
         torch.cuda.empty_cache()
-        expanded = expand_prompt(prompt+caption+caption_2)
+        expanded = expand_prompt(caption)
         expanded_1 = expanded[0]
         expanded_2 = expanded[1]
-
-        expanded_1 = flatten_and_stringify(expanded_1)
-        expanded_1 = " ".join(expanded_1)
-        expanded_2 = flatten_and_stringify(expanded_2)
-        expanded_2 = " ".join(expanded_2)
-        
-        prompt = flatten_and_stringify(prompt+expanded_1+expanded_2)
-        prompt = " ".join(prompt)
-        
+        new_prompt = prompt+' '+expanded_1
+        print("-- ------------ --")
+        print("-- FINAL PROMPT --")
+        print(new_prompt)
+        print("-- FINAL PROMPT --")
+        print("-- ------------ --")
         global model
         global txt_tokenizer
         del model
@@ -723,18 +725,15 @@ def generate_90(
         del processor5
         gc.collect()
         torch.cuda.empty_cache()
-        expanded = expand_prompt(prompt+caption+caption_2)
+        expanded = expand_prompt(caption)
         expanded_1 = expanded[0]
         expanded_2 = expanded[1]
-
-        expanded_1 = flatten_and_stringify(expanded_1)
-        expanded_1 = " ".join(expanded_1)
-        expanded_2 = flatten_and_stringify(expanded_2)
-        expanded_2 = " ".join(expanded_2)
-        
-        prompt = flatten_and_stringify(prompt+expanded_1+expanded_2)
-        prompt = " ".join(prompt)
-        
+        new_prompt = prompt+' '+expanded_1
+        print("-- ------------ --")
+        print("-- FINAL PROMPT --")
+        print(new_prompt)
+        print("-- FINAL PROMPT --")
+        print("-- ------------ --")
         global model
         global txt_tokenizer
         del model