debug pipeline

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")
+    chatbot = pipeline("text-generation", model="mistralai/Mistral-7B-Instruct-v0.3", max_new_tokens=2048,)
     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},
@@ -83,7 +83,8 @@ with gr.Blocks(css=css) as demo:
             show_label=False,
             max_lines=10, 
             placeholder="Prompt refined by Mistral", 
-            container=False
+            container=False,
+            max_length=2048,
             )