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" in input:
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"