Update app.py

app.py

@@ -298,7 +298,7 @@ def expand_prompt(prompt):
         outputs = model.generate(
             input_ids=input_ids,
             attention_mask=attention_mask,
-            max_new_tokens=256,
+            max_new_tokens=128,
             temperature=0.2,
             top_p=0.9,
             do_sample=True,
@@ -312,7 +312,7 @@ def expand_prompt(prompt):
         outputs_2 = model.generate(
             input_ids=input_ids_2,
             attention_mask=attention_mask_2,
-            max_new_tokens=384,
+            max_new_tokens=128,
             temperature=0.2,
             top_p=0.9,
             do_sample=True,
@@ -423,13 +423,10 @@ def generate_30(
         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]
-
-        prompt = flatten_and_stringify(prompt+expanded_1+expanded_2)
-        prompt = " ".join(prompt)
-        
+        new_prompt = prompt+' '+expanded_1
         global model
         global txt_tokenizer
         del model
@@ -448,7 +445,7 @@ def generate_30(
                 pil_image_3=sd_image_c,
                 pil_image_4=sd_image_d,
                 pil_image_5=sd_image_e,
-                prompt=prompt+' '+expanded_1,
+                prompt=new_prompt,
                 negative_prompt=negative_prompt,
                 text_scale=text_scale,
                 ip_scale=ip_scale,