ChirathD
/

Blip-2-test-1

Visual Question Answering

Inference Endpoints

8-bit precision

Model card Files Files and versions Community

ChirathD commited on Dec 6, 2023

Commit

67b0b4a

•

1 Parent(s): 5815286

Update handler.py

Files changed (1) hide show

handler.py +2 -1

handler.py CHANGED Viewed

@@ -4,7 +4,8 @@ from PIL import Image
 import torch
 import os
 from io import BytesIO
-from transformers import BlipForConditionalGeneration, BlipProcessor
 # -
 device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')

 import torch
 import os
 from io import BytesIO
+# from transformers import BlipForConditionalGeneration, BlipProcessor
+from transformers import Blip2Processor, Blip2ForConditionalGeneration
 # -
 device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')