debugging

inference.py

@@ -10,7 +10,7 @@ class Inference:
         self.logger = logging.get_logger("transformers")
 
     def inference(self, selected, image, text):
-        self.logger.info(f"selected model {selected}, question {text.value}")
+        self.logger.info(f"selected model {selected}, question {text}")
         if selected == "Blip Saffal":
             return self.__inference_saffal_blip(image, text)
         elif selected == "Blip CN":
@@ -22,10 +22,12 @@ class Inference:
         encoding = self.blip_processor(image, text, return_tensors="pt")
         out = self.blip_model_saffal.generate(**encoding, max_new_tokens=100)
         generated_text = self.blip_processor.decode(out[0], skip_special_tokens=True)
+        self.logger.info(f"answer {generated_text}")
         return f"{generated_text}"
 
     def __inference_control_net_blip(self, image, text):
         encoding = self.blip_processor(image, text, return_tensors="pt")
         out = self.blip_model_control_net.generate(**encoding, max_new_tokens=100)
         generated_text = self.blip_processor.decode(out[0], skip_special_tokens=True)
+        self.logger.info(f"answer {generated_text}")
         return f"{generated_text}"