remove kwargs

app.py

@@ -44,7 +44,7 @@ def get_text_gen_pipeline():
                    model="mistralai/Mistral-7B-Instruct-v0.3",
                    max_new_tokens=2048,
                    device=device,
-                   model_kwargs={"add_prefix_space": False}
+                   #model_kwargs={"add_prefix_space": False}
               )
          except Exception as e:
               print(f"Error loading text generation model: {e}")
@@ -58,7 +58,7 @@ def refine_prompt(prompt):
         return "Text generation model is unavailable."
     try:
         messages = [
-            {"role": "system", "content": "You are a product designer. You will get a basic prompt of product request and you need to imagine a new product design to satisfy that need. Produce an extended description of product front view that will be used by a text-to-image AI (Flux) to generate a visual. The prompt should be in the form of a product description, not a story. Mention the visual aesthetics. Background should be a white background."},
+            {"role": "system", "content": "You are a product designer. You will get a basic prompt of product request and you need to imagine a new product design to satisfy that need. Produce an extended description of product front view that will be used by a text-to-image AI (Flux) to generate a visual. The prompt should be in the form of a product description, not a story, maximum 2048 tokens. Mention the visual aesthetics. Background should be a white background."},
             {"role": "user", "content": prompt},
         ]
         refined_prompt = text_gen(messages)