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

543ffeb

•

1 Parent(s): 67b0b4a

Update handler.py

Files changed (1) hide show

handler.py +1 -1

handler.py CHANGED Viewed

@@ -13,7 +13,7 @@ device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
 class EndpointHandler():
     def __init__(self, path=""):
         # load the optimized model
         self.processor = Blip2Processor.from_pretrained("ChirathD/Blip-2-test-1")
         self.model = Blip2ForConditionalGeneration.from_pretrained("ChirathD/Blip-2-test-1").to(device)
         self.model.eval()

 class EndpointHandler():
     def __init__(self, path=""):
         # load the optimized model
+        print("####### Start Deploying #####")
         self.processor = Blip2Processor.from_pretrained("ChirathD/Blip-2-test-1")
         self.model = Blip2ForConditionalGeneration.from_pretrained("ChirathD/Blip-2-test-1").to(device)
         self.model.eval()