nougat fix

1 year ago · 633c5a1d71
parent 0fcd63d1fc
commit 633c5a1d71
1 changed files with 1 additions and 3 deletions
--- 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(