increase length

app.py

@@ -21,7 +21,7 @@ MAX_SEED = np.iinfo(np.int32).max
 MAX_IMAGE_SIZE = 2048
 
 def refine_prompt(prompt):
-    chatbot = pipeline("text-generation", model="mistralai/Mistral-7B-Instruct-v0.3", max_new_tokens=2048,)
+    chatbot = pipeline("text-generation", model="mistralai/Mistral-7B-Instruct-v0.3", max_new_tokens=2048, device=device)
     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 use by Flux to generate a visual"},
         {"role": "user", "content": prompt},
@@ -40,7 +40,8 @@ def infer(prompt, seed=42, randomize_seed=False, width=1024, height=1024, num_in
             height = height,
             num_inference_steps = num_inference_steps, 
             generator = generator,
-            guidance_scale=0.0
+            guidance_scale=0.0,
+            max_sequence_length=2048
     ).images[0] 
     return image, seed
  
@@ -190,7 +191,7 @@ with gr.Blocks(css=css) as demo:
     gr.on(
         triggers=[run_button.click],
         fn = infer,
-        inputs = [prompt, seed, randomize_seed, width, height, num_inference_steps],
+        inputs = [refined_prompt, seed, randomize_seed, width, height, num_inference_steps],
         outputs = [result, seed]
     )