sheraz179 commited on
Commit
48a12f4
·
1 Parent(s): dc8c69d

Update handler.py

Browse files
Files changed (1) hide show
  1. handler.py +2 -2
handler.py CHANGED
@@ -12,9 +12,9 @@ class EndpointHandler():
12
  def __init__(self, path=""):
13
  # load the optimized model
14
 
15
- self.processor = Blip2Processor.from_pretrained("Salesforce/blip-image-captioning-base")
16
  self.model = Blip2ForConditionalGeneration.from_pretrained(
17
- "Salesforce/blip-image-captioning-base"
18
  ).to(device)
19
  self.model.eval()
20
  self.model = self.model.to(device)
 
12
  def __init__(self, path=""):
13
  # load the optimized model
14
 
15
+ self.processor = Blip2Processor.from_pretrained("Salesforce/blip2-flan-t5-xl")
16
  self.model = Blip2ForConditionalGeneration.from_pretrained(
17
+ "Salesforce/blip2-flan-t5-xl"
18
  ).to(device)
19
  self.model.eval()
20
  self.model = self.model.to(device)