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
|