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"
|