sheraz179
/

blip2-flan-t5-xl

image-captioning

visual-question-answering

Model card Files Files and versions

sheraz179 commited on Jul 7, 2023

Commit

48a12f4

·

1 Parent(s): dc8c69d

Update handler.py

Files changed (1) hide show

handler.py +2 -2

handler.py CHANGED Viewed

@@ -12,9 +12,9 @@ class EndpointHandler():
     def __init__(self, path=""):
         # load the optimized model
-        self.processor = Blip2Processor.from_pretrained("Salesforce/blip-image-captioning-base")
         self.model = Blip2ForConditionalGeneration.from_pretrained(
-            "Salesforce/blip-image-captioning-base"
         ).to(device)
         self.model.eval()
         self.model = self.model.to(device)

     def __init__(self, path=""):
         # load the optimized model
+        self.processor = Blip2Processor.from_pretrained("Salesforce/blip2-flan-t5-xl")
         self.model = Blip2ForConditionalGeneration.from_pretrained(
+            "Salesforce/blip2-flan-t5-xl"
         ).to(device)
         self.model.eval()
         self.model = self.model.to(device)