diff --git a/swarms/models/nougat.py b/swarms/models/nougat.py index 453c6cae..aedb6be4 100644 --- a/swarms/models/nougat.py +++ b/swarms/models/nougat.py @@ -61,9 +61,7 @@ class Nougat: def __call__(self, img: str): """Call the model with an image_path str as an input""" image = Image.open(img) - pixel_values = self.processor( - image, return_tensors="pt" - ).pixel_values + pixel_values = self.processor(image, return_tensors="pt").pixel_values # Generate transcriptions, here we only generate 30 tokens outputs = self.model.generate(