Update app.py

app.py

@@ -122,6 +122,7 @@ def load_and_prepare_model():
         text_encoder=None,
         text_encoder_2=None,
         vae=None,
+        unet=None,
     )
     
     '''
@@ -294,7 +295,7 @@ def expand_prompt(prompt):
         outputs = model.generate(
             input_ids=input_ids,
             attention_mask=attention_mask,
-            max_new_tokens=384,
+            max_new_tokens=512,
             temperature=0.2,
             top_p=0.9,
             do_sample=True,
@@ -307,7 +308,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=512,
             temperature=0.2,
             top_p=0.9,
             do_sample=True,
@@ -411,6 +412,8 @@ def generate_30(
         expand_prompt(prompt)
         expand_prompt(caption)
         expanded = expand_prompt(caption_2)
+        expanded_1 = expanded[0]
+        expanded_2 = expanded[1]
         del model
         del txt_tokenizer
         gc.collect()
@@ -426,8 +429,8 @@ def generate_30(
                 pil_image_3=sd_image_c,
                 pil_image_4=sd_image_d,
                 pil_image_5=sd_image_e,
-                prompt=prompt,
-                prompt_2=expanded,
+                prompt=prompt+' '+expanded_1,
+                prompt_2=expanded_2,
                 negative_prompt=negative_prompt,
                 text_scale=text_scale,
                 ip_scale=ip_scale,