sooh-j
/

blip2-vizwizqa

Visual Question Answering

Inference Endpoints

Model card Files Files and versions Community

sooh-j commited on Jun 4, 2024

Commit

d6ea99d

·

verified ·

1 Parent(s): a6def05

Update handler.py

Files changed (1) hide show

handler.py +1 -1

handler.py CHANGED Viewed

@@ -3,7 +3,7 @@
 # import logging
 # import copy
 import numpy as np
-from transformers import Blip2Processor, Blip2ForConditionalGeneration, BlipForQuestionAnswering
 from typing import Dict, List, Any
 from PIL import Image
 from transformers import pipeline

 # import logging
 # import copy
 import numpy as np
+from transformers import Blip2Processor, Blip2ForConditionalGeneration, BlipForQuestionAnswering, BitsAndBytesConfig
 from typing import Dict, List, Any
 from PIL import Image
 from transformers import pipeline