Spaces:
Build error
Build error
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(
|