Update app.py

app.py

@@ -126,11 +126,11 @@ Your task:
   "shoes": "High-quality product image of brown leather boots, white background",
   "hat": "Studio photo of a navy beret on a stand, isolated on white"
 }
-Respond only with a valid JSON object.
+Respond only with a valid JSON object. Ensure the JSON is properly formatted with correct commas between fields.
+Do not forget commas between entries. Validate before finishing your response.
 Do not include any explanations or markdown syntax. No commentary. No extra text.
 Start directly with `{` and end with `}`.
-Ensure the JSON is properly formatted with correct commas between fields.
-Do not forget commas between entries. Validate before finishing your response.
+
 """
 
     prompt = f"""<|system|>
@@ -139,7 +139,7 @@ Do not forget commas between entries. Validate before finishing your response.
 "{outfit_sentence}"</s>
 """
 
-    outputs = pipe(prompt, max_new_tokens=512, temperature=0.7, top_p=0.9)
+    outputs = pipe(prompt, max_new_tokens=512, temperature=0.7, top_k=50, top_p=0.9)
     pattern = r'\<\|system\|\>(.*?)\<\|assistant\|\>'
     cleaned_text = re.sub(pattern, '', outputs[0]["generated_text"], flags=re.DOTALL)
     print(f"\n🧾 Raw LLM response:\n{cleaned_text}")