advaitadasein
/

blip2_test

visual-question-answering

image-captioning

Inference Endpoints

Model card Files Files and versions Community

advaitadasein commited on Mar 6

Commit

e9fb176

•

1 Parent(s): ff92cea

Update handler.py

Files changed (1) hide show

handler.py +2 -2

handler.py CHANGED Viewed

@@ -8,8 +8,8 @@ from io import BytesIO
 class EndpointHandler():
     def __init__(self, path=""):
-        self.processor = Blip2Processor.from_pretrained("Salesforce/blip2-opt-6.7b")
-        self.model = Blip2ForConditionalGeneration.from_pretrained("Salesforce/blip2-opt-6.7b")
         self.device = "cuda" if torch.cuda.is_available() else "cpu"
         self.model.to(self.device)

 class EndpointHandler():
     def __init__(self, path=""):
+        self.processor = Blip2Processor.from_pretrained("Salesforce/blip2-opt-2.7b")
+        self.model = Blip2ForConditionalGeneration.from_pretrained("Salesforce/blip2-opt-2.7b")
         self.device = "cuda" if torch.cuda.is_available() else "cpu"
         self.model.to(self.device)