added debugging prints

app.py

@@ -91,6 +91,8 @@ async def generate_endpoint(
     available = list(MODELS.keys())
     model_selection = random.choices(available, k=5)
 
+    print("Selected models: ", model_selection)
+
     for m_name in model_selection:
         m_data = MODELS[m_name]
         model = m_data["model"]
@@ -99,7 +101,7 @@ async def generate_endpoint(
             i_proc, t_proc = m_data["processor"]
             pixel_values = i_proc(images=image, return_tensors="pt").pixel_values.to(DEVICE)
             gen_ids = model.generate(
-                pixel_values=pixel_values, max_length=200, do_sample=True, 
+                pixel_values=pixel_values, max_length=300, do_sample=True, 
                 temperature=temp, top_k=top_k, top_p=top_p
             )
             cap = t_proc.batch_decode(gen_ids, skip_special_tokens=True)[0]
@@ -107,13 +109,15 @@ async def generate_endpoint(
             proc = m_data["processor"]
             pixel_values = proc(images=image, return_tensors="pt").pixel_values.to(DEVICE)
             gen_ids = model.generate(
-                pixel_values=pixel_values, max_length=200, do_sample=True, 
+                pixel_values=pixel_values, max_length=300, do_sample=True, 
                 temperature=temp, top_k=top_k, top_p=top_p
             )
             cap = proc.batch_decode(gen_ids, skip_special_tokens=True)[0]
         
         captions.append(cap.strip())
 
+        print("Caption generated: ", cap.strip())
+
     return {"captions": captions, "mix": model_selection}
 
 @app.post("/ui-tester")