diff --git a/backend.py b/backend.py index b34af5a..d4fb28c 100644 --- a/backend.py +++ b/backend.py @@ -13,7 +13,7 @@ print(torch.cuda.is_available()) # checkpoint = "sam_vit_b_01ec64.pth" print(time.time()) # set large model related configs -checkpoint = "sam_vit_h_4b8939.pth" +checkpoint = "notebooks/sam_vit_h_4b8939.pth" model_type = "vit_h" sam = sam_model_registry[model_type](checkpoint=checkpoint) sam.to(device='cuda') @@ -54,6 +54,9 @@ def index(): return response else: return 'Content-Type not supported!' - - -print('server starts working') \ No newline at end of file + + +print('server starts working') + + +app.run('0.0.0.0', 2020, False)