Update app.py

app.py

@@ -93,19 +93,15 @@ def unified_chat(image, message, chat_history, seed, top_p, temperature, cfg_wei
         if chat_history and len(chat_history) > 0:
             # Get the last few turns of conversation for context (limit to last 3 turns)
             recent_context = chat_history[-3:] if len(chat_history) > 3 else chat_history
-            context_text = " ".join([f"{user}: {user_msg}" for user_msg, _ in recent_context])
+            context_text = " ".join([f"User: {user_msg}" for user_msg, _ in recent_context])
+            #context_text = " ".join([f"{user}: {user_msg}" for user_msg, _ in recent_context])
             
             # Only use context if it's not too long
             if len(context_text) < 200:  # Arbitrary length limit
                 context_prompt = f"{context_text}. {extracted_prompt}"
         
         # Generate images
-        generated_images = generate_image(
-            prompt=context_prompt,
-            seed=seed,
-            guidance=cfg_weight,
-            t2i_temperature=t2i_temperature
-        )
+        generated_images = generate_image(prompt=context_prompt, seed=seed, guidance=cfg_weight, t2i_temperature=t2i_temperature)
         
         # Create a response that includes the generated images
         response = f"I've generated the following images based on: '{extracted_prompt}'"
@@ -247,7 +243,7 @@ def generate_image(prompt, seed=None, guidance=5, t2i_temperature=1.0, progress=
         np.random.seed(seed)
     width = 384
     height = 384
-    parallel_size = 4
+    parallel_size = 1
     
     with torch.no_grad():
         messages = [{'role': '<|User|>', 'content': prompt},