Spaces:
Build error
Build error
Update app.py
Browse files
app.py
CHANGED
@@ -48,11 +48,9 @@ if not os.path.exists(pretrained_pth_novg):
|
|
48 |
build model
|
49 |
'''
|
50 |
model_last = BaseModel(opt, build_model(opt)).from_pretrained(pretrained_pth_last).eval().cuda()
|
51 |
-
model_cap = BaseModel(opt, build_model(opt)).from_pretrained(pretrained_pth_novg).eval().cuda()
|
52 |
|
53 |
with torch.no_grad():
|
54 |
model_last.model.sem_seg_head.predictor.lang_encoder.get_text_embeddings(["background", "background"], is_eval=True)
|
55 |
-
model_cap.model.sem_seg_head.predictor.lang_encoder.get_text_embeddings(["background", "background"], is_eval=True)
|
56 |
|
57 |
'''
|
58 |
inference model
|
|
|
48 |
build model
|
49 |
'''
|
50 |
model_last = BaseModel(opt, build_model(opt)).from_pretrained(pretrained_pth_last).eval().cuda()
|
|
|
51 |
|
52 |
with torch.no_grad():
|
53 |
model_last.model.sem_seg_head.predictor.lang_encoder.get_text_embeddings(["background", "background"], is_eval=True)
|
|
|
54 |
|
55 |
'''
|
56 |
inference model
|