pdich2085
/

new-blip

image-captioning

endpoints-template

Inference Endpoints

Model card Files Files and versions Community

pdich2085 commited on Oct 14, 2023

Commit

a7af85b

·

1 Parent(s): 55095dc

Update handler.py

Files changed (1) hide show

handler.py +1 -1

handler.py CHANGED Viewed

@@ -7,7 +7,7 @@ from transformers import BlipForConditionalGeneration, BlipProcessor
 device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
 class EndpointHandler():
-    def __init__(self):
         self.processor = BlipProcessor.from_pretrained("Salesforce/blip-image-captioning-large")
         self.model = BlipForConditionalGeneration.from_pretrained(
             "Salesforce/blip-image-captioning-large"

 device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
 class EndpointHandler():
+    def __init__(self, path=""):
         self.processor = BlipProcessor.from_pretrained("Salesforce/blip-image-captioning-large")
         self.model = BlipForConditionalGeneration.from_pretrained(
             "Salesforce/blip-image-captioning-large"