Update model.py

model.py

@@ -381,8 +381,8 @@ class StreamMultiDiffusion(nn.Module):
         question = 'Question: What are in the image? Answer:'
         inputs = self.i2t_processor(image, question, return_tensors='pt')
         out = self.i2t_model.generate(**{k: v.to(self.i2t_model.device) for k, v in inputs.items()}, max_new_tokens=77)
-        print(out)
         prompt = self.i2t_processor.decode(out[0], skip_special_tokens=True).strip()
+        print(prompt)
         return prompt
 
     @torch.no_grad()