Spaces:
Runtime error
Runtime error
farrosalferro24
commited on
Commit
β’
75fa56b
1
Parent(s):
07d53ab
Update model/processing_gecko.py
Browse files
model/processing_gecko.py
CHANGED
@@ -149,7 +149,7 @@ class GeckoProcessor(ProcessorMixin):
|
|
149 |
|
150 |
assert len(processed_images) == blocks + 1
|
151 |
|
152 |
-
return self.image_processor(images=processed_images, return_tensors='pt'
|
153 |
|
154 |
|
155 |
def preprocess_interleaved_images_and_text(
|
|
|
149 |
|
150 |
assert len(processed_images) == blocks + 1
|
151 |
|
152 |
+
return self.image_processor(images=processed_images, return_tensors='pt')['pixel_values'], coords
|
153 |
|
154 |
|
155 |
def preprocess_interleaved_images_and_text(
|