Update app_v2.py

app_v2.py

@@ -75,10 +75,9 @@ def caption(input_image: Image.Image, prompt: str, temperature: float, top_p: fl
         streamer=streamer,
     )
     _= cap_model.generate(**generate_kwargs)
-    outputs = []
-    for text in streamer:
-        outputs.append(text)
-        yield "".join(outputs)
+
+    output = cap_model.generate(**generate_kwargs)
+    print(f"Generated {len(output[0])} tokens")
 
 @spaces.GPU()
 @torch.no_grad()