Update handler.py
Browse files- handler.py +1 -1
handler.py
CHANGED
@@ -13,7 +13,7 @@ device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
|
|
13 |
class EndpointHandler():
|
14 |
def __init__(self, path=""):
|
15 |
# load the optimized model
|
16 |
-
|
17 |
self.processor = Blip2Processor.from_pretrained("ChirathD/Blip-2-test-1")
|
18 |
self.model = Blip2ForConditionalGeneration.from_pretrained("ChirathD/Blip-2-test-1").to(device)
|
19 |
self.model.eval()
|
|
|
13 |
class EndpointHandler():
|
14 |
def __init__(self, path=""):
|
15 |
# load the optimized model
|
16 |
+
print("####### Start Deploying #####")
|
17 |
self.processor = Blip2Processor.from_pretrained("ChirathD/Blip-2-test-1")
|
18 |
self.model = Blip2ForConditionalGeneration.from_pretrained("ChirathD/Blip-2-test-1").to(device)
|
19 |
self.model.eval()
|