Ankur Goyal
commited on
Commit
·
02daaaf
1
Parent(s):
3f1b351
Small arg handling fixes
Browse files
pipeline_document_question_answering.py
CHANGED
|
@@ -221,7 +221,7 @@ class DocumentQuestionAnsweringPipeline(Pipeline):
|
|
| 221 |
|
| 222 |
image = None
|
| 223 |
image_features = {}
|
| 224 |
-
if "image"
|
| 225 |
if not VISION_LOADED:
|
| 226 |
raise ValueError(
|
| 227 |
"If you provide an image, then the pipeline will run process it with PIL (Pillow), but"
|
|
|
|
| 221 |
|
| 222 |
image = None
|
| 223 |
image_features = {}
|
| 224 |
+
if input.get("image", None) is not None:
|
| 225 |
if not VISION_LOADED:
|
| 226 |
raise ValueError(
|
| 227 |
"If you provide an image, then the pipeline will run process it with PIL (Pillow), but"
|