Update app.py

app.py

@@ -327,13 +327,13 @@ def captioning(img, prompt_1,prompt_2,prompt_3,prompt_4,prompt_5):
         generated_ids = model5.generate(
             **inputs,
             do_sample=True,
-            num_beams=1,
+            num_beams=5,
             max_length=64,
             #min_length=16,
             top_p=0.1,
             repetition_penalty=1.0,
             length_penalty=2.0,
-            temperature=0.5,
+            temperature=0.55,
         )
         generated_text = processor5.batch_decode(generated_ids, skip_special_tokens=True)[0].strip()
         response_text = generated_text.replace(prompt, "").strip() #Or could try .split(prompt, 1)[-1].strip()