updating the inference

inference.py

@@ -20,12 +20,12 @@ class Inference:
 
     def __inference_saffal_blip(self, image, text):
         encoding = self.blip_processor(image, text, return_tensors="pt")
-        out = self.blip_model_saffal.generate(**encoding)
+        out = self.blip_model_saffal.generate(**encoding, max_new_tokens=100)
         generated_text = self.blip_processor.decode(out[0], skip_special_tokens=True)
         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)
+        out = self.blip_model_control_net.generate(**encoding, max_new_tokens=100)
         generated_text = self.blip_processor.decode(out[0], skip_special_tokens=True)
         return f"{generated_text}"