bugs fixes

src/model.py

@@ -37,6 +37,8 @@ class Pixtral:
         ]
 
         outputs = self.llm.chat(messages, sampling_params=self.sampling_params)
+        print("OUTPUT")
+        print(outputs[0].outputs[0].text)
 
         return outputs[0].outputs[0].text
 

src/pipelines.py

@@ -144,6 +144,9 @@ class InvoiceGenerator:
 
         print("Detailed Damage Description:", detailed_damage_description)
 
+        print("PROMPT")
+        print(GENERATE_INVOICE_PROMPT(invoice_info, detailed_damage_description))
+
         generated_invoice = self.model.generate_message_from_image(
             GENERATE_INVOICE_PROMPT(invoice_info, detailed_damage_description), image_path
         ).replace("```markdown", "").replace("```", "")