From 633c5a1d716818cb9c8e6ad08305a6016dd76c6a Mon Sep 17 00:00:00 2001
From: Kye <kye@apacmediasolutions.com>
Date: Fri, 24 Nov 2023 15:24:32 -0800
Subject: [PATCH] nougat fix

---
 swarms/models/nougat.py | 4 +---
 1 file changed, 1 insertion(+), 3 deletions(-)

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(